“蔚来杯“2022牛客暑期多校训练营5 Birds in the tree

最新推荐文章于 2024-10-09 22:28:58 发布

wantBe_better

最新推荐文章于 2024-10-09 22:28:58 发布

阅读量107

点赞数

文章标签：算法

本文链接：https://blog.csdn.net/wantbe_better/article/details/126120571

版权

本文解析了如何使用树形动态规划求解一个给定的编程题目，涉及一棵节点带颜色的树，目标是找到度数为1的节点颜色相同且相连的子树的不同方案数。通过递推公式展示了如何计算以每个节点为根的子树中符合条件的方案数，并提供了关键代码实现。

摘要由CSDN通过智能技术生成

题面：https://ac.nowcoder.com/acm/contest/33190/D
分析：
题意相当于给你一棵树，树上的节点有两种颜色，求联通子树中度数为1的点颜色相同的方案数
对于这一道题，我们可以想到用树形DP来求解
我们设 $dp_{i,col}$ 为以 $i$ 为根的子树内 $x$ 被选择,除 $x$ 外叶节点为 $co l$ 的方案数
在这里插入图片描述可以看出当颜色为 $co l$ 时 $dp_{x,col}=\prod_{s\epsilon son_x}(dp_{son_{x,col}}+1)$
对 $dp_{son_{x,col}}+1$ 的解释：因为对于每一个方案可以取 $0至dp_{son_{x,col}}$ ，所以共可取 $dp_{son_x}+1$ 种

代码:

#include<bits/stdc++.h>

using namespace std;

const int MXN=3*1e5+7;
const long long mod=1e9+7;


int n;
vector<int> g[MXN];
long long dp[MXN][3];
long long res=0;
char s[MXN];

inline int read(){
	int x=0,f=1;
	char ch=getchar();
	while(ch<'0'||ch>'9'){
		ch=getchar();
	}
	while(ch>='0'&&ch<='9'){
		x=(x<<3)+(x<<1)+(ch^48);
		ch=getchar();
	}
	return x;
}


void dfs(int u,int fa){
	dp[u][1]=dp[u][0]=1;
	int i,v;
	long long s1=0,s0=0;
	for(i=0;i<g[u].size();i++){
		v=g[u][i];
		if(v==fa) continue;
		dfs(v,u);
		dp[u][1]=(1ll*dp[u][1]*(dp[v][1]+1))%mod;
		dp[u][0]=(1ll*dp[u][0]*(dp[v][0]+1))%mod;
		s1=(s1+dp[v][1])%mod;
		s0=(s0+dp[v][0])%mod;
	}
	if(s[u]=='1'){
		dp[u][0]--;
		res=(res+dp[u][1])%mod;
		res=(res-s0+dp[u][0])%mod; 
	} else {
		dp[u][1]--;
		res=(res+dp[u][0])%mod;
		res=(res-s1+dp[u][1])%mod;
	}
}
int main(){
	int i,x,y;
	n=read();
	scanf("%s",s+1);
	for(i=1;i<n;i++){
		x=read(); y=read();
		g[x].push_back(y);
		g[y].push_back(x);
	}
	dfs(1,-1);
    printf("%lld\n",(res%mod+mod)%mod);
}