题目:
输入一个n个节点的无根树的各条边,并指定一个根节点,要求把该树转化为有根树,输出各个节点的父亲编号。
分析:分析在代码的注释中!
#include <iostream> #include <vector> using namespace std; const int MAXN = 1000; int n, p[MAXN]; vector<int> G[MAXN]; void dfs(int u, int fa) { //递归转化为以u为根的子树,u的父亲为fa int d = G[u].size(); //节点u的相邻点的个数 for(int i = 0; i < d; ++i) { //循环遍历跟这个节点相连接的d个节点。 int v = G[u][i]; //节点u的第i个相邻点v if(fa != v) dfs(v, p[v] = u); //把v的父亲节点设为u,然后递归转化为以v为根的子树 //一定要判断v是否和其父亲节点相等! } } int main() { cin >> n; for(int i = 0; i < n-1; i++) { //输入n-1条边 int u, v; cin >> u >> v; G[u].push_back(v); G[v].push_back(u); } int root; cin >> root; //指定根节点。 p[root] = -1; //设定根节点的父亲节点为-1,代表根节点没有父亲节点。 dfs(root, -1); for(int i = 0; i < n; ++i) { cout << p[i] << endl; } return 0; }
ACM:树的变换,无根树转有根树
时间: 2024-10-07 00:28:25