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

1 #include<cstdio>
2 #include<vector>
3 #include<stack>
4 #define FOR(a,b,c) for(int a=(b);a<(c);a++)
5 using namespace std;
6
7 const int maxn= 1000 + 10;
8
9 int pos[maxn];      //BFS中的位置关系
10 vector<int> G[maxn];
11
12 int main(){
13 int n;
14   while(scanf("%d",&n)==1){
15         int x;
16       FOR(i,1,n+1){
17           scanf("%d",&x);
18           pos[x]=i;
19           G[i].clear();     //G_clear
20       }
21       int root;
22       scanf("%d",&root);
23       stack<int> sta; sta.push(root);
24       int v;
25       FOR(i,1,n){
26           scanf("%d",&v);
27           for(;;){
28               int u=sta.top();
29               if(u==root || pos[u]+1<pos[v]){  //u是v的父结点
30                   sta.push(v);
31                   G[u].push_back(v);
32                   break;            //直到找到父结点
33               }
34               else  //u和v是兄弟 返回到u的父结点
35                sta.pop();
36           }
37       }
38       FOR(u,1,n+1){
39           printf("%d:",u);
40           FOR(j,0,G[u].size()) printf(" %d",G[u][j]);
41         printf("\n");
42       }
43   }
44   return 0;
45 }