PTA L3-029 还原文件 (30 分)

PTA L3-029 还原文件 (30 分)

#include<bits/stdc++.h>
using namespace std;

const int N = 100010;

vector<int> g[N],res;
bool st[N];
int a[N];

int n,m;

void dfs(int d)
{
	if(d>=n-1)
	{
		cout<<res[0]+1;
		for(int i=1;i<res.size();i++)
			cout<<" "<<res[i]+1;
	}
	else
	{
		for(int i=0;i<m;i++)
			if(!st[i])
			{
				bool ok=0;
				for(int j=0;j<g[i].size();j++)
					if(g[i][j]!=a[d+j])
					{
						ok=1;
						break;
					}
				if(!ok)
				{
					st[i]=1;
					res.push_back(i);
					dfs(d+g[i].size()-1);
					st[i]=0;
					res.pop_back();
				}
			}
	}
}

int main()
{
	cin>>n;
	for(int i=0;i<n;i++) cin>>a[i];
	
	cin>>m;
	for(int i=0;i<m;i++)
	{
		int k;cin>>k;
		for(int j=0;j<k;j++)
		{
			int x;cin>>x;
			g[i].push_back(x);
		}
	}
	
	dfs(0);
	
	return 0;
}