习题6-11 树重建 UVa 10410 *

题意：输入n节点树的bfs序列和dfs序列。输出每个节点的子节点列表，其中输入序列是这样生成的：当一个节点被扩展时，其所有子节点应该按照编号从小到大的顺序被访问。

Sample Input

8
4 3 5 1 2 8 7 6
4 3 1 7 2 6 5 8

Sample Output

1: 7
2: 6
3: 1 2
4: 3 5
5: 8
6:
7:
8:

分析：又没写出来，还是太菜了。看到题解就立马明白了，先放一下，以后再看看。

#include<cstdio>
#include<vector>
#include<cstring>
using namespace std;
const int maxn=1010;
vector<int> tree[maxn];
int pos[maxn],dfs[maxn],n;
int main()
{
freopen("f.txt","r",stdin);
while(~scanf("%d",&n)){
memset(pos,0,sizeof(pos));
memset(dfs,0,sizeof(dfs));
int k,p,q=0;
for(int i=0;i<n;i++){
scanf("%d",&k);
pos[k]=i;
tree[i].clear();
}
scanf("%d",&dfs[0]);
for(int i=1;i<n;i++){
scanf("%d",&p);
while(q&&pos[p]<=pos[dfs[q]]+1)--q;
tree[dfs[q]-1].push_back(p);
dfs[++q]=p;
}
for(int i=0;i<n;i++){
printf("%d:",i+1);
for(int j=0;j<tree[i].size();j++){
printf(" %d",tree[i][j]);
}
printf("\n");
}
}
return 0;
}