YbtOJ 状压DP问题课堂过关 例3 涂抹果酱【状压DP】

题目

思路

这道题我们要用三进制处理就好了。
注意先要筛出合法方案。

代码

#include<iostream>
#include<cstdio>
using namespace std;
const int mod=1000000;
int n,m,k,a[100],num,eve[300],tot,v[300][300],ans;
int f[10010][300],statu[300][300],bj,kid;
int main()
{
	scanf("%d%d",&n,&m);
	scanf("%d",&k);
	for(int i=1; i<=m; i++)
	 {
	   scanf("%d",&a[i]);
	   a[i]--;
	 }
	int total=1;
	for(int i=1; i<=m; i++)
	   total=total*3;
	for(int i=0; i<total; i++)
	 {
	 	int num=i,js=0;
		bj=0;
	 	while(num)
	 	 {
	 	 	eve[++js]=num%3;
	 	 	num/=3;
		 }
		for(int j=2; j<=m; j++)
		 {
		 	if(eve[j]==eve[j-1])
		 	 {
		 	 	bj=1;
		 	 	break;
			 }
		 }
		if(bj==0)
		 {
		 	++tot;
		 	for(int j=1; j<=m; j++)
		 	   statu[tot][j]=eve[j];
		 }
	 }
	for(int i=1; i<=tot; i++)
	 for(int j=1; j<=tot; j++)
	  {
	  	 bj=0;
	  	 for(int k=1; k<=m; k++)
	  	  {
	  	  	if(statu[i][k]==statu[j][k])
	  	  	 {
	  	  	 	bj=1;
	  	  	 	break;
			 }
		  }
		 if(bj==1)
		   v[i][j]=v[j][i]=0;
		 else
		   v[i][j]=v[j][i]=1;
	  }
	for(int i=1; i<=tot; i++)
	 {
	 	bj=0;
        for(int j=1; j<=m; j++)
         {
         	if(a[j]!=statu[i][j])
         	 {
         	 	bj=1;
         	 	break;
			 }
		 }
		if(bj==0)
		 {
		 	kid=i;
		 	break;
		 }
	 }
	if(k==1)
	 {
	 	f[1][kid]=1;
	 	for(int i=2; i<=n; i++)
	 	 {
	 	 	for(int j=1; j<=tot; j++)
	 	 	 for(int k=1; k<=tot; k++)
	 	 	  {
	 	 	  	if(v[j][k])
	 	 	  	  f[i][j]=(f[i][j]%mod+f[i-1][k]%mod)%mod;
			  }
		 }
	 }
	else
	 {
	 	for(int i=1; i<=tot; i++)
	 	   f[1][i]=1;
	 	for(int i=1; i<=n; i++)
	 	 {
	 	 	if(i==k)
	 	 	 {
	 	 	 	for(int j=1; j<=tot; j++)
	 	 	  	 if(v[kid][j])
	 	 	       f[i][kid]=(f[i][kid]%mod+f[i-1][j]%mod)%mod;
	 	 	    continue;
			 }
	 	 	for(int j=1; j<=tot; j++)
	 	 	 for(int k=1; k<=tot; k++)
	 	 	  {
	 	 	  	if(v[j][k])
	 	 	  	  f[i][j]=(f[i][j]%mod+f[i-1][k]%mod)%mod;
			  }
		 }
	 }
	if(k!=n)
	 {
	    for(int i=1; i<=tot; i++)
	       ans=(ans%mod+f[n][i]%mod)%mod;
	    printf("%d",ans);
	 }
	else
	  printf("%d",f[n][kid]%mod);
	return 0;
}