UVa 10410 Tree Reconstruction 树重建 构造 栈

题目链接：Tree Reconstruction
题目描述：

给定一个$n$个结点的树的$DFS$序与$BFS$序（$DFS$和$BFS$的过程如下：当一个结点有多个结点时，优先遍历编号较小的节点），输入任意一种满足条件树的每个结点的子结点列表。

题解：

给定一个$DFS$序和$BFS$序并不一定能够唯一确定一棵树，我们只需要构造出一棵树让他的$BFS$序和$DFS$序与给定而相同即可。
那么如何构造呢？由$BFS$的性质可以知道，某个节点在$BFS$序中的位置即代表该结点到根节点的距离。而构造方法可以借助一个栈来完成，首先需要将根节点加入栈中，然后我们依次遍历$DFS$序，我们比较当前遍历的结点到根节点的距离$d_u$与栈顶元素到根节点的距离$d_t$：

• 如果$t$是根节点，那么$u$一定是$t$的子节点，我们将$t$入栈
• 再如果$d_u-d_t\le 1$，我们此时只需要弹出栈顶元素，继续拿栈中下一个元素进行比较。
• 再如果$d_u-d_t>1$，此时我们直接将$u$作为$t$的儿子节点，同时将$t$入栈。

上述方法为什么是正确的呢？最让人疑惑的实际上是上面条件的第三点，为什么满足条件三就是父子关系，而不可能是兄弟关系呢？实际上，如果按照上述的流程，是不可能出现栈顶的元素与遍历的结点不是相邻的兄弟关系，这是因为如果$t$和$u$是兄弟，但是并不相邻，那么我们可以假设$t$和$u$之间有另一个兄弟结点$v$，可以知道这三者在$BFS$序中出现的顺序应该是$t->v->u$，这意味着在遍历到$u$的时候,$v$一定已经遍历过了，那么按照前两条判断条件，如果$t,v$是兄弟关系，那么$t$一定会被弹出，而$u$会被保留在栈中，也就是说栈中实际上保存的每一层最靠右的结点。所以并不会发生$t$与$u$是相隔开的兄弟的情况。

代码：

#include <bits/stdc++.h>

const int MAXN = 1010;

using namespace std;

int n, x, root;
int dis[MAXN];
vector<int> ans[MAXN];
stack<int> s;

int main()
{
    ios::sync_with_stdio(false);
    while(cin >> n) {
        for (int i = 0; i < n; i++) {
            cin >> x;
            dis[x] = i;
            ans[i + 1].clear();
        }
        stack<int>().swap(s);
        for (int i = 0; i < n; i++) {
            cin >> x;
            if (i == 0) {
                root = x;
                s.push(x);
                continue;
            }
            while(s.top() != root && dis[x] - dis[s.top()] <= 1) { s.pop(); }
            ans[s.top()].push_back(x);
            s.push(x);
        }
        for (int i = 1; i <= n; i++) {
            cout << i << ": ";
            for (auto son : ans[i]) {
                cout << son << " ";
            }
            cout << endl;
        }

    }
    return 0;
}