L3-029 还原文件 (30 分)-PAT 团体程序设计天梯赛 GPLT

一份重要文件被撕成两半，其中一半还被送进了碎纸机。我们将碎纸机里找到的纸条进行编号，如图 1 所示。然后根据断口的折线形状跟没有切碎的半张纸进行匹配，最后还原成图 2 的样子。要求你输出还原后纸条的正确拼接顺序。

输入格式：
输入首先在第一行中给出一个正整数 N（1<N≤105），为没有切碎的半张纸上断口折线角点的个数；随后一行给出从左到右 N 个折线角点的高度值（均为不超过 100 的非负整数）。

随后一行给出一个正整数 M（≤100），为碎纸机里的纸条数量。接下去有 M 行，其中第 i 行给出编号为 i（1≤i≤M）的纸条的断口信息，格式为：
K h[1] h[2] … h[K]
其中 K 是断口折线角点的个数（不超过 104 +1），后面是从左到右 K 个折线角点的高度值。为简单起见，这个“高度”跟没有切碎的半张纸上断口折线角点的高度是一致的。

输出格式：
在一行中输出还原后纸条的正确拼接顺序。纸条编号间以一个空格分隔，行首尾不得有多余空格。题目数据保证存在唯一解。

输入样例：
17
95 70 80 97 97 68 58 58 80 72 88 81 81 68 68 60 80
6
4 68 58 58 80
3 81 68 68
3 95 70 80
3 68 60 80
5 80 72 88 81 81
4 80 97 97 68

输出样例：
3 6 1 5 2 4

分析：h[i]存储第i个折线角点的高度，frag[i]存储第i个纸条碎片的信息。通过DFS从没有切碎的纸片最左端开始尝试所有纸条碎片的匹配，p表示当前所需要匹配的碎片左侧在未碎纸片上的位置，将匹配上的纸条碎片编号添加到Ans中记录，并使用vis标记是否已经匹配上了，如果Ans的容量等于纸片数量，则表示已经完成所有匹配，输出结果并返回～

#include <bits/stdc++.h>
using namespace std;
int n, m, k, t, flag1, flag2;
int h[100001], vis[101];
vector<int> Ans, frag[101];
void DFS (int p) {
    if ((int)Ans.size() == m) {
        flag1 = 1;
        for (int i = 0; i < m; i++) {
            if (i) cout << ' ';
            cout << Ans[i];
        }
    }
    if (flag1) return;
    for (int i = 1; i <= m; i++) {
        if (vis[i]) continue;
        flag2= 0;
        for (int j = 0; j < (int)frag[i].size(); j++) {
            if (frag[i][j] != h[p + j]) {
                flag2 = 1;
                break;
            }
        }
        if (flag2) continue;
        Ans.push_back(i);
        vis[i] = 1;
        DFS(p + (int)frag[i].size() - 1);
        vis[i] = 0;
        Ans.pop_back();
    }
};
int main() {
    cin >> n;
    for (int i = 0; i < n; i++) cin >> h[i];
    cin >> m;
    for (int i = 1; i<= m; i++) {
        cin >> k;
        for (int j = 0; j < k; j++) {
            cin >> t;
            frag[i].push_back(t);
        }
    }
    DFS(0);
    return 0;
}

#include <bits/stdc++.h>

using namespace std;

int n, m, k, t, flag1, flag2;

int h[100001], vis[101];

vector<int> Ans, frag[101];

void DFS (int p) {

if ((int)Ans.size() == m) {

flag1 = 1;

for (int i = 0; i < m; i++) {

if (i) cout << ' ';

cout << Ans[i];

}

if (flag1) return;

for (int i = 1; i <= m; i++) {

if (vis[i]) continue;

flag2= 0;

for (int j = 0; j < (int)frag[i].size(); j++) {

if (frag[i][j] != h[p + j]) {

flag2 = 1;

break;

}

if (flag2) continue;

Ans.push_back(i);

vis[i] = 1;

DFS(p + (int)frag[i].size() - 1);

vis[i] = 0;

Ans.pop_back();

}

};

int main() {

cin >> n;

for (int i = 0; i < n; i++) cin >> h[i];

cin >> m;

for (int i = 1; i<= m; i++) {

cin >> k;

for (int j = 0; j < k; j++) {

cin >> t;

frag[i].push_back(t);

}

DFS(0);

return 0;

}

❤ 点击这里 -> 订阅《PAT | 蓝桥 | LeetCode学习路径 & 刷题经验》by 柳婼

❤ 点击这里 -> 订阅《从放弃C语言到使用C++刷算法的简明教程》by 柳婼

❤ 点击这里 -> 订阅PAT甲级乙级、蓝桥杯、GPLT天梯赛、LeetCode题解离线版