DFS一遍求子树的节点数

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf = 0x3f3f3f3f;
const int maxn = 1e5+50;
vector<int>g[maxn];
int n;
int dfs(int u,int f)
{
	int sum = 1;
	for(int i=0;i<g[u].size();i++)
	{
		int v = g[u][i];
		if(v!=f)
		{
			sum += dfs(v,u);
		}
	}
	printf("(节点%d的子树节点数:%d)\n",u,sum);
	return sum;
}
int main()
{
	cin>>n;
	for(int i=1;i<=n-1;i++)
	{
		int u,v; cin>>u>>v;
		g[u].push_back(v);
		g[v].push_back(u);
	}
	dfs(1,0);
}
/*
4
1 2
2 3
2 4
*/

posted @ 2020-07-01 20:46  特立独行的猪猪  阅读(307)  评论(0编辑  收藏  举报