PTA 还原文件

题目

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

图1纸条编号

图2还原结果

输入

输入首先在第一行中给出一个正整数 N（$1<N\le 10^5$），为没有切碎的半张纸上断口折线角点的个数；随后一行给出从左到右 N 个折线角点的高度值（均为不超过 100 的非负整数）。
随后一行给出一个正整数 M（≤100），为碎纸机里的纸条数量。接下去有 M 行，其中第 i 行给出编号为 i（1≤i≤M）的纸条的断口信息，格式为：

K h[1] h[2] ... h[K]

其中 K 是断口折线角点的个数（不超过 $10^4+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

题目分析

此题可按照纸条的缺口逐一寻找对应的纸条，但由于可能存在两张纸条为{70,70}{70,70,70},那么前者可能会误占后者的位置，造成答案错误，因此本题还需要回溯步骤，直至判定位置移至纸条末尾。

代码

#include<iostream>
#include<vector>
using namespace std;
int a[100005],ans[105],flag[105];
vector<int> paper[105];//存储各碎纸条
int n,m,t=-1;

void find(int begin){//判定是否吻合并且得到下一次判定的首地址
	if(begin >= n) return ;
	for(int j = 1;j<=m;++j){//逐个比较
		if(flag[j]==0){//标记已放置的纸条
			int status = 0,i = begin;
			for(int k = 0;k<paper[j].size();++k){
				if(a[i++]!=paper[j][k]){
					status = 1;//当前纸条不匹配
					break;
				}
			}
			if(status == 0){
				ans[++t]=j;
				flag[j]=1;
				find(begin+paper[ans[t]].size()-1);
                --t;//回溯
                flag[j]= 0;
			}
		}
	}
	
}

int main(){
	cin >> n;
	for(int i = 1;i<=n;++i)
		scanf("%d",&a[i]);
	cin >> m;
	for(int i = 1;i<=m;++i){
		int l;
		cin >> l;
		for(int j = 0;j<l;++j){
			int k;
			scanf("%d",&k);
			paper[i].push_back(k);
		}
	}
	find(1);
	for(int i = 0;i<m-1;++i)
		cout << ans[i]<<" ";
	cout << ans[m-1];
	return 0;
}

运行结果