题解多边形染色

Posted 2021-02-20 h-lka

Link

蒟蒻做的为数不多的环形(dp),技巧不到位，写题解来总结一下。

( ext{Solution:})

(dp)柿子还是很好推出来的：(dp[i][j])表示地(i)个点染色是(j)的方案数。先考虑没有限制的转移：

[dp[i][j]=sum_{k ot=j} dp[i-1][k] ]

最后注意一下(1,n)的限制。

这题多了三个操作，分别是：限制相邻两个颜色一样，限制固定颜色，以及限制不能填的颜色。

那么我们分类讨论一下，对于有相邻颜色一样的点，就可以从上一步的相同颜色转移而来，同时注意一下是不是有其他两种操作即可。

对于没有相邻颜色限制的点，考虑其他限制，有固定颜色的枚举上一个所有不同于固定颜色的颜色转移；有不能填的颜色的枚举上一个所有颜色特判一下就行。

这题的主要难点在于环形后效性的处理。对于我这种没怎么写过环形(dp)的人，还是有点生疏。

对于这个题，瞪眼法看出最后一个颜色与第一个颜色是有关联的。那么我们可以钦定第一个点的颜色，进行(c)次(dp)就可以方便地处理后效性啦。

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int mod=987654321;
int n,m,c,ans;
int col[50010],dp[50010][11];
//dp[i]表示i这个点是p颜色，1-i一共的方案数
int vis[50010],sum[50010];
int dislike[50010][20];
signed main() {
	scanf("%lld%lld%lld",&n,&m,&c);
	for(int i=1; i<=m; ++i) {
		int opt,x,y;
		scanf("%lld%lld%lld",&opt,&x,&y);
		if(opt==1)for(int i=1;i<=c;++i){if(i!=y)dislike[x][i]=1;}
		if(opt==2)dislike[x][y]=1;
		if(opt==3) {
			if(y>x)swap(x,y);
			vis[x]=y;
		}
	}
	//vis[big]=small
	/*if(col[1]==0){
		for(int i=1;i<=c;++i)dp[1][i]=1;
	}
	if(col[1]!=0){
		if(col[1]>0)dp[1][col[1]]=1;
		else{
			for(int i=1;i<=c;++i){
				if(i==-col[1])continue;
				dp[1][i]=1;
			}
		}
	}
	for(int i=1;i<=n;++i){
		if(!vis[i])continue;
		if(col[i]<=0)continue;
		col[vis[i]]=col[i];
	}
	for(int i=1;i<=c;++i)sum[1]+=dp[1][i];
	for(int i=2;i<=n;++i){
		if(col[i]==0&&!vis[i]){
			//无限制
			for(int j=1;j<=c;++j){
				dp[i][j]*=(sum[i-1]-dp[i-1][j]);
			}
		}
		else if(col[i]==0&&vis[i]){

		}
	}*/
	/*if(col[1]==0)for(int i=1;i<=c;++i)dp[1][i]=1;
	else if(col[1]>0)dp[1][col[1]]=1;
	else for(int i=1;i<=c;++i){if(i!=-col[1])dp[1][i]=1;}
	for(int i=2;i<=n;++i){
		if(vis[i]){
			if(col[i]==0){
				for(int j=1;j<=c;++j)dp[i][j]=dp[i-1][j];
			}
			else{
				if(col[i]<0){
					for(int j=1;j<=c;++j){
						if(j==-col[i])continue;
						dp[i][j]=dp[i-1][j];
					}
				}
				else{
					dp[i][col[i]]=dp[i-1][col[i]];
				}
			}
		}
		else{
			//not same&&col is free
			if(col[i]==0){
				for(int j=1;j<=c;++j){
					for(int k=1;k<=c;++k){
						if(j==k)continue;
						dp[i][j]+=dp[i-1][k];
					}
				}
				//cout<<"qwq
";
			}
			else if(col[i]<0){
				for(int j=1;j<=c;++j){
					if(j==-col[i])continue;
					for(int k=1;k<=c;++k){
						if(j==k)continue;
						dp[i][j]+=dp[i-1][k];
					}
				}
			}
			else{
				for(int j=1;j<=c;++j){
					if(j==col[i])continue;
					dp[i][col[i]]+=dp[i-1][j];
				}
			}
		}
	}*/
	for(int p=1; p<=c; ++p) {
		//one‘s color
		if(-col[1]==p)continue;
		for(int i=1; i<=n; ++i)for(int j=1; j<=c; ++j)dp[i][j]=0;//prepare for
		dp[1][p]=1;//first
		for(int i=2; i<n; ++i) {
			
			for(int j=1; j<=c; ++j) {
				if(dislike[i][j])continue;
				if(vis[i]) {
					dp[i][j]+=dp[i-1][j];
					dp[i][j]%=mod;
				} else {
					for(int k=1; k<=c; ++k) {
						if(k==j)continue;
						dp[i][j]+=dp[i-1][k];
						dp[i][j]%=mod;
					}
				}
			}
		}
			for(int i=1; i<=c; ++i) {
				if(dislike[n][i]||i==p)continue;
				if(vis[n])dp[n][i]+=dp[n-1][i],dp[n][i]%=mod;
				else {
					for(int j=1; j<=c; ++j) {
						if(i==j)continue;
						dp[n][i]+=dp[n-1][j];
						dp[n][i]%=mod;
					}
				}
			}
		for(int i=1; i<=c; ++i)ans+=dp[n][i],ans%=mod;
	}
	printf("%lld
",ans%mod);
	return 0;
}

注意题目细节，操作(1,2)并不一定只有一个，所以要开一个数组来存储所有不能染的颜色，就是代码中的(dislike[][].)

看代码中思路不清晰的后果，注释了一大堆，写了一百五十多行……