Tree Reconstruction UVA - 10410 (BFS，DFS反推，栈)

题目链接

题意：

给出一颗树的bfs和dfs，注意不一定是二叉树！！

输出每个结点的子结点。

分析：

明白bfs可以描述结点与根结点的距离关系。用一个栈来维护，每次将根结点入栈，如果下一个结点到根结点的距离大于栈顶元素到根结点的距离，则该结点为栈顶元素的子结点。

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1000+10;
int dis[maxn];
vector<int> Node[maxn];
int main() {
    freopen("i.txt","r",stdin);
    freopen("o.txt","w",stdout);
    int n,t,root;
    while(cin >> n && n) {
        memset(dis, 0, sizeof(dis));
        for(int i = 1; i <= n; i++) {
            Node[i].clear();
            cin >> t;
            dis[t] = i-1;
        }
        stack<int> stack1;
        cin >> root; stack1.push(root);
        for(int i = 1; i < n; i++) {
            cin >> t;
            while(true) {
                int u = stack1.top();
                if(root==u || dis[t]>dis[u]+1) {
                    Node[u].push_back(t);
                    stack1.push(t);
                    break;
                }
                else stack1.pop();
            }
        }
        for(int i = 1; i <= n; i++) {
            cout << i << ":";
            for(int j = 0; j < (int)Node[i].size(); j++)
                cout << " " << Node[i][j];
            cout << endl;
        }
    }
    return 0;
}