UVa 10410 - Tree Reconstruction

最新推荐文章于 2022-10-09 23:28:27 发布

原创最新推荐文章于 2022-10-09 23:28:27 发布 · 536 阅读

0 ·

CC 4.0 BY-SA版权

UVA 专栏收录该内容

232 篇文章

订阅专栏

本文详细阐述了使用栈来维护节点之间的距离关系，通过BFS序列映射每个节点到根节点的距离，并利用DFS遍历来记录节点间的父子关系。重点解析了在比较节点距离时的特殊情况，即当节点距离根节点比栈顶节点大1时，视为相等，以准确识别兄弟节点。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include<cstdio>
#include<cstring>
#include<stack>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn = 1005;

int N, pos[maxn];
vector<int> g[maxn];

int main(){
    while(scanf("%d",&N)==1){
        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(true){

                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;
}

就是不知道为什么pos[u]+1<pos[x]

解题思路：用栈维护即可。对应BFS序列映射出了每个节点和根节点的距离，遍历dfs序列，对当前节点和栈顶节点比较，如果该节点距离根节点更远，则说明该节点为栈顶节点个孩子节点，则记录后将节点放入栈中。否则弹掉栈顶元素继续比较。需要注意一点，即当元素与栈顶元素的距离值大1的时候要视为相等,因为它们属于兄弟节点