【树形dp】洛谷P3914 树染色题解

最新推荐文章于 2024-06-13 14:43:16 发布

搂鱼114514

最新推荐文章于 2024-06-13 14:43:16 发布

阅读量183

点赞数 7

文章标签：动态规划算法

本文链接：https://blog.csdn.net/m0_60506105/article/details/138437229

版权

速切蓝题，但是石门拓展版却不对？

题目 $f(u,i)=s(v)-f(v,i)$

染色计数 - 洛谷

有一颗N个节点的树，节点用1,2,⋯ ,N编号。你要给它染色，使得相邻节点的颜色不同。M种颜色，用1,2,⋯ ,M编号。每个节点可以染M种颜色中的若干种，求不同染色方案的数量除以1e9+7的余数。

思路

因为限定了颜色，所以先标记，设 $f(u,x)=1$ 表示u节点可以(1)染颜色x

再考虑树上，对于树上的相邻点对(u,v)、u是v的父亲，当知道了v的染色方案数时，就用总共有的颜色减去已染的颜色就可以得到u的染色方案数

于是设 $s(u)=\sum_{i=1}^{m}f(u,i)$ ，那么易得状态转移方程：

$f(u,i)=s(v)-f(v,i)$

在dfs的处理：

void dfs(ll u,ll fa)
{
	for(int i=head[u];i;i=e[i].next)
	{
		ll v=e[i].to;
		if(v==fa)continue;
		dfs(v,u);
		for(int j=1;j<=m;j++)
		{
			ll able=(s[v]-f[v][j]+mod)%mod;
			f[u][j]=f[u][j]*able%mod;
		}
	}
	for(int j=1;j<=m;j++)
	s[u]=(s[u]+f[u][j])%mod;
}

代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const ll N=5005,mod=1e9+7;
ll n,m,k,x,u,v;
ll f[N][N],s[N];
//f[i][x]:点i可染颜色x
//s[i]=Σf[i][1~m] 
ll idx,head[N<<1];
struct edge
{
	ll to,next;
}e[N<<1];
void addedge(ll u,ll v)
{
	idx++;
	e[idx].to=v;
	e[idx].next=head[u];
	head[u]=idx;
}
void dfs(ll u,ll fa)
{
	for(int i=head[u];i;i=e[i].next)
	{
		ll v=e[i].to;
		if(v==fa)continue;
		dfs(v,u);
		for(int j=1;j<=m;j++)
		{
			ll able=(s[v]-f[v][j]+mod)%mod;
			f[u][j]=f[u][j]*able%mod;
		}
	}
	for(int j=1;j<=m;j++)
	s[u]=(s[u]+f[u][j])%mod;
}
int main()
{
	scanf("%lld%lld",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%lld",&k);
		for(int j=1;j<=k;j++)
		{
			scanf("%lld",&x);
			f[i][x]=1;
		}
	}
	for(int i=1;i<n;i++)
	{
		scanf("%lld%lld",&u,&v);
		addedge(u,v);
		addedge(v,u);
	}
	dfs(1,0);
	printf("%lld",s[1]);
	return 0;
}