图论专题小结:无根树转有根树

来源:互联网 发布:人马一体 知乎 编辑:程序博客网 时间:2024/05/22 01:51

1.题意:输入一个n个结点的无根树的各条边,并指定一个根结点,要求把该树转化为有根树,输出各个节点的父节点编号。n≤10^6。

2.思路:用vector数组存放每个结点的相邻结点,vector与n成正比。

#define N 1000000+10vector<int>G[N];int p[N];int n;void read_tree(){int u, v;scanf("%d", &n);for (int i = 0; i < n - 1; i++){scanf("%d%d", &u, &v);G[u].push_back(v);G[v].push_back(u);}}void dfs(int u, int fa){int d = G[u].size();for (int i = 0; i < d; i++){int v = G[u][i];if (v != fa)//注意此处判断,否则会无限递归dfs(v, p[v] = u);}}int main(){read_tree();int root;cin >> root;p[root] = -1;dfs(root, -1);for (int i = 0; i < n; i++)printf("%d%c", p[i], i == n - 1 ? '\n' : ' ');return 0;}

0 0
原创粉丝点击