Uva-10410 Tree Reconstruction

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

Minakami_Yuki

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

阅读量537

点赞数

CC 4.0 BY-SA版权

分类专栏：数据结构文章标签：树栈

本文链接：https://blog.youkuaiyun.com/Minakami_Yuki/article/details/48328915

数据结构专栏收录该内容

3 篇文章

订阅专栏

本文详细介绍了如何利用DFS序和BFS序重建一棵树的算法。通过先进行BFS构建节点映射，再利用DFS序和栈来确定树的结构，特别强调了兄弟节点的定义和唯一性问题。代码示例提供了实现细节。

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

题目链接：Tree Reconstruction

题目大意：给定一棵树的DFS序和BFS序，求这棵树。

解题思路：先跑一遍BFS，并将每个节点映射成BFS序，由于BFS的特性，其序号越大，离根结点的距离相等或更远。接着遍历DFS序，并用栈维护结点之间的关系，若当前序号比栈顶元素大，则该结点为孩子结点，否则pop掉栈顶元素继续与下一个元素比较。由于只告诉BFS和DFS序所建立的树并不唯一，所以我们强制认为只比当前结点序号大1的结点为该结点的兄弟结点。

代码如下：

#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cstdio>
#include <vector>
#include <cstdlib>
#include <cstring>
#include <iomanip>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef pair<int, int> P;
const int inf = 0x3f3f3f3f;
const int maxn = 2e4 + 15;

int pos[1005];

int main(){
    ios::sync_with_stdio(false);
    #ifdef LOCAL
    freopen("in.txt", "r", stdin);
    #endif
    int n, node, root;
    while(cin >> n){
        vector<int> chd[1005];
        for(int i = 1; i <= n; i++){
            cin >> node;
            pos[node] = i;
        }       
        cin >> node;
        stack<int> sta;
        sta.push(node);
        root = node;
        for(int i = 2; i <= n; i++){
            cin >> node;
            while(1){
                int e = sta.top();
                if(e == root || pos[node] > pos[e] + 1){
                    chd[e].push_back(node);
                    sta.push(node);
                    break;
                }
                else
                    sta.pop();
            }
        }
        for(int i = 1; i <= n; i++){
            cout << i << ":";
            for(auto e : chd[i]){
                cout << ' ' << e;   
            }
            cout << endl;
        }
    }
    return 0;
}