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

UVa10410 Tree Reconstruction

算法：根据BFS构造pos数组以区分关系，在此基础上对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=  + ;

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

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