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