uva 10410 - Tree Reconstruction

题目链接

题目大意：给一棵树的bfs结果，和dfs结果，求各节点的子节点（从小到大）

解题思路：根据BFS序列存每个节点和根节点的距离，

遍历dfs序列，对当前节点和栈顶节点比较，

如 该节点距离根节点更远，则该节点为栈顶节点个孩子节点，

则记录后将节点放入栈中。

否则弹掉栈顶元素继续比较。

注意事项：vector 要清空

距离为1时是兄弟结点

#include<cstdio>
#include<cstring>
#include<stack>
#include<vector>
#include<algorithm>
using namespace std;

int n, pos[1005];    //记录节点与根节点的距离
vector<int> g[1005];

int main()
{
    while(scanf("%d",&n)!=EOF)
    {
            int x;
        for(int i=1;i<=n;i++)
        {
            scanf("%d", &x);
            g[i].clear();
            pos[x]=i;
        }

        int root;
        stack<int> sta;          //存储父亲节点
        scanf("%d", &root);
        sta.push(root);

        for(int i=1;i<n;i++)
        {
            scanf("%d", &x);

            while(1)           //判断是否是子节点或者兄弟节点
            {

                int u=sta.top();    //当前节点

                if(u==root || pos[u]+1 < pos[x])    //兄弟结点
                {
                    g[u].push_back(x);
                    sta.push(x);
                    break;
                }
                else
                    sta.pop();         //如果两节点距离相邻则呼唤父亲节点。栈的下一个节点是当前栈顶节点的父节点
            }
        }

        for(int i=1;i<=n;i++)
        {
            printf("%d:",i);
            for(int j=0;j<g[i].size();j++)
                printf(" %d",g[i][j]);
            printf("\n");
        }
    }
    return 0;
}