“今日头条杯”首届湖北省大学程序设计竞赛 D. Who killed Cock Robin

最新推荐文章于 2021-05-01 11:50:32 发布

原创最新推荐文章于 2021-05-01 11:50:32 发布 · 341 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#水题 #图论

随笔专栏收录该内容

39 篇文章

订阅专栏

本文介绍了一种使用树形动态规划解决特定问题的方法——计算一棵树的所有子树的总数。通过定义状态dp[u]表示以u为根节点的子树数量，并利用递归思想更新状态，最终得到整棵树的所有子树总数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

D. Who killed Cock Robin

题目链接

题意：给出一颗树，求它的所有子树的总数。

思路：定义一个dp[],dp[u]是指以u为结点的子树的个数。

可以推出：对于u所有的直接子节点v,dp[u]=dp[u]*(dp[v]+1);

可以理解为（dp[v]+1）是在v结点形成的所有子树中加入一个u结点这样形成的一定含有u结点的子树总数。

#include<bits/stdc++.h>
using namespace std;
const int maxn = 200000 + 10;
const int INF=int(1e9);
const int mod=10000007;
#define ll long long
vector<int>edge[maxn];
bool vis[maxn];
ll dp[maxn];
void init(int n) {
	for(int i=0; i<=n; i++)
		edge[i].clear();
	memset(vis,0,sizeof(vis));

}
void dfs(int u) {
	vis[u]=true;
	dp[u]=1;
	for(int i=0; i<edge[u].size(); i++) {
		int v=edge[u][i];
		if(vis[v])
			continue;
		dfs(v);
		dp[u]=dp[u]*(dp[v]+1)%mod;
	}
	return;
}
int main() {
	int n;
	int u,v;
	scanf("%d",&n);
	init(n);
	for(int i=1; i<=n-1; i++) {
		scanf("%d %d",&u,&v);
		edge[u].push_back(v);
		edge[v].push_back(u);
	}
	dfs(1);
	ll ans=0;
	for(int i=1; i<=n; i++)
		ans=(ans+dp[i])%mod;
	printf("%lld\n",ans);
	return 0;
}