无根树转化为有根树

1. #include <iostream>  
2. #include <vector>  
3. using namespace std;  
4.   
5. const int MAXN = 1000;  
6. int n, p[MAXN];  
7. vector<int> G[MAXN];  
8.   
9. void dfs(int u, int fa) {   //递归转化为以u为根的子树，u的父亲为fa  
10.     int d = G[u].size();        //节点u的相邻点的个数  
11.     for(int i = 0; i < d; ++i) {    //循环遍历跟这个节点相连接的d个节点。  
12.         int v = G[u][i];       //节点u的第i个相邻点v  
13.         if(fa != v) dfs(v, p[v] = u);  //把v的父亲节点设为u，然后递归转化为以v为根的子树  
14.         //一定要判断v是否和其父亲节点相等！  
15.     }  
16. }  
17.   
18. int main() {  
19.     cin >> n;  
20.     for(int i = 0; i < n-1; i++) {   //输入n-1条边  
21.         int u, v;  
22.         cin >> u >> v;  
23.         G[u].push_back(v);  
24.         G[v].push_back(u);  
25.     }  
26.     int root;     
27.     cin >> root;    //指定根节点。  
28.     p[root] = -1;   //设定根节点的父亲节点为-1，代表根节点没有父亲节点。  
29.     dfs(root, -1);  
30.     for(int i = 0; i < n; ++i) {  
31.         cout << p[i] << endl;  
32.     }  
33.     return 0;  
34. }  

 

 

 

[cpp] view plain copy