URAL - 1069 Prufer Code (图的重建)

给定一个树,每次删除一个编号最小的叶子节点,然后将叶子节点的父亲节点输出,要求根据该序列还原这棵树。

思路:找度为0的点,(这样的点不会成为父亲节点,只能是叶子节点)放到优先队列中,模拟删除点的过程即可。(没删除一个点,就对应序列中的一个父亲节点)

代码如下:

#include <cstdio>
#include <queue>
#include <vector>
#include <algorithm>
using namespace std;
#define M 7505
priority_queue<int, vector<int>, greater<int> >q;
int n, du[M], a[M], x;
vector<int>g[M];
int main ()
{
    n = 1;

    while(~scanf("%d",&x))
    {
        a[n] = x;
        ++du[x];
        ++n;
    }
    for(int i = 1; i <= n; ++i)
        if(du[i]==0) q.push(i);
    for(int i = 1; i < n; ++i)
    {
        x = q.top(); q.pop();
        g[a[i]].push_back(x);
        g[x].push_back(a[i]);
        --du[a[i]];
        if(du[a[i]]==0) q.push(a[i]);
    }
    for(int i = 1; i <= n; ++i)
    {
        printf("%d:",i);
        sort(g[i].begin(), g[i].end());
        for(int j = 0; j < (int)g[i].size(); ++j)
            printf(" %d",g[i][j]);
        printf("\n");
    }
    return 0;
}


你可能感兴趣的:(URAL - 1069 Prufer Code (图的重建))