您的位置:首页 > 其它

dfs(顺便求一下每个节点的儿子节点喽)

2016-03-07 20:32 453 查看
先用邻接表存储这棵树(图),(**邻接表相当于链表^_^),然后搜寻每个节点,递归进它的子树,然后就没有然后了……

顺便提供数据O(∩_∩)O哈!

12

1 2 1

1 3 1

1 4 1

3 5 1

3 6 1

4 7 1

5 8 1

6 9 1

7 10 1

8 11 1

10 12 1

3 0 2 1 1 1 1 1 0 1 0 0

#include<iostream>
#include<cstring>
using namespace std;
const int mn=10000+10,me=2*mn+10;
int hd[mn],to[me],nxt[me],w[me],cnt;
void add(int x,int y,int z){
to[cnt]=y;
w[cnt]=z;
nxt[cnt]=hd[x];
hd[x]=cnt++;
}
int st[mn];//st[]代表每个节点的子树数目
bool vis[mn];//记录每个点是否被访问过,避免重复访问
void dfs(int i){
vis[i]=true;
for(int j=hd[i];j!=-1;j=nxt[j]){
if(!vis[to[j]]){
dfs(to[j]);//递归呀递归^_^
st[i]++;
}
}
}
int main(){
int n;//n个点
cin>>n;
cnt=0;
memset(hd,-1,sizeof(hd));
memset(to,0,sizeof(to));
memset(nxt,0,sizeof(nxt));
memset(vis,false,sizeof(vis));
memset(st,0,sizeof(st));
for(int i=1,x,y,z;i<n;i++){
cin>>x>>y>>z;
add(x,y,z);
}
dfs(1);
for(int i=1;i<=n;i++)
cout<<st[i]<<" ";
cout<<endl;
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  dfs