【Ybtoj 第24章例3】涂抹果酱【状压DP】

---

解题思路

注意要状压成 $3$ 进制 因为有 $3$ 种颜色。$f_{i,j}$ 表示第 i行状态为j的方案总数

已经涂好了第 $k$行 那就分别从 $k-1$到 $0$ 和$k+1$到 n ，dp 两个方案数相乘 就是答案了，特判涂第 1 行 和第 n 行的情况。

---

代码

#include<iostream>
#include<cstdio>
#include<cmath>
#include<iomanip>

using namespace std;

const int Mod=1e6;
int n,m,k,a,p,z,len,K,num,st[300];
long long m1,m2,f[10010][300];

bool check(int x)
{
	int a=0x3f;
	for(int i=1;i<=m;i++)
	{
		if(a==x%3)
		 	return 0;
		a=x%3,x=x/3;
	}
	return 1;
}

bool ch(int x,int y)
{
	for(int i=1;i<=m;i++)
	{
		if((x%3)==(y%3))	
			return 0;
		x=x/3,y=y/3;
	}
	return 1;
}

int main(){
	scanf("%d%d%d",&n,&m,&K);
	for(int i=1;i<=m;i++)
	{
		scanf("%d",&a);
		p=p*3+(a-1);
	}
	len=pow(3,m);
	for(int i=0;i<len;i++)
	{
		if(check(i))
			st[++num]=i;;
	}
	for(int i=1;i<=num;i++)
		if(st[i]==p)
		{
			z=i;
			break;
		}
	if(!z)
	{
		cout<<0;
		return 0;
	}
	f[K][z]=1;
	for(int i=K-1;i>=1;i--)	
		for(int j=1;j<=num;j++)	
			for(int k=1;k<=num;k++)
				{
					if(ch(st[j],st[k]))
						f[i][j]=(f[i][j]+f[i+1][k])%Mod;
				}
		
	for(int i=K+1;i<=n;i++)
		for(int j=1;j<=num;j++)
			for(int k=1;k<=num;k++)
					if(ch(st[j],st[k]))
					{
						f[i][j]=(f[i][j]+f[i-1][k])%Mod;
					}	
	for(int i=1;i<=num;i++)
		m1=(m1+f[1][i])%Mod,m2=(m2+f[n][i])%Mod;
	if(K==1)
		printf("%lld",m2%Mod);
	else if(K==n)
		printf("%lld",m1%Mod);
	else printf("%lld",(m1*m2)%Mod);
	
}