【暑假】[基本数据结构]根据BFS与DFS确定树

#include<cstdio>
#include<vector>
#include<stack>
#define FOR(a,b,c) for(int a=(b);a<(c);a++)
using namespace std;

const int maxn= 1000 + 10;

int pos[maxn];      //BFS中的位置关系
vector<int> G[maxn];

int main(){
int n;
while(scanf("%d",&n)==1){
int x;
FOR(i,1,n+1){
scanf("%d",&x);
pos[x]=i;
G[i].clear();     //G_clear
}
int root;
scanf("%d",&root);
stack<int> sta; sta.push(root);
int v;
FOR(i,1,n){
scanf("%d",&v);
for(;;){
int u=sta.top();
if(u==root || pos[u]+1<pos[v]){  //u是v的父结点
sta.push(v);
G[u].push_back(v);
break;            //直到找到父结点
}
else  //u和v是兄弟 返回到u的父结点
sta.pop();
}
}
FOR(u,1,n+1){
printf("%d:",u);
FOR(j,0,G[u].size()) printf(" %d",G[u][j]);
printf("\n");
}
}
return 0;
}