牛客练习赛97-月之暗面-(树形dp)

D

题意：
就是给你一个n点的树，有x种普通颜色，y种特殊颜色，现在要给树上每个相邻的节点染色，普通的颜色没有限制，但是相邻的两个不能然相同的特殊颜色，问你方案数，同时取模。

思考：
其实一看就知道就是经典的树上模型，但是好像脑子有点傻逼，不知道怎么下手去写？每个点的方案数就是看看他的父亲就行了，但是这样写下来，怎么求总的方案数呢，直接乘?不对的。其实就是直接从下往上来递归求，和以前做过的一道题其实也很像，就是多思考思考这种类型就可以了。

代码：

int T,n,m,k;
int a,b;
int va[N];
int dp[N][2];

vector<int > e[N];

void dfs(int now,int p)
{
	dp[now][0] = dp[now][1] = 1;
	for(auto spot:e[now])
	{
		if(spot==p) continue;
		dfs(spot,now);
		dp[now][0] = dp[now][0]*(dp[spot][0]*a%mod+dp[spot][1]*b%mod)%mod;
		dp[now][1] = dp[now][1]*(dp[spot][0]*a%mod+dp[spot][1]*(b-1)%mod)%mod;
	}
}

signed main()
{
	IOS;
	cin>>n>>a>>b;
	for(int i=1;i<n;i++)
	{
		int a,b;
		cin>>a>>b;
		e[a].pb(b);
		e[b].pb(a);
	}
	dfs(1,0);
	int ans = dp[1][0]*a%mod+dp[1][1]*b%mod;
	cout<<ans%mod<<"\n";
	return 0;
}

总结：
多多思考呀。