一份重要文件被撕成两半,其中一半还被送进了碎纸机。我们将碎纸机里找到的纸条进行编号,如图 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的容量等于纸片数量,则表示已经完成所有匹配,输出结果并返回~
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
#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; } |