涂抹果酱【状压dp】

题目描述

Tyvj 两周年庆典要到了，Sam 想为 Tyvj 做一个大蛋糕。蛋糕俯视图是一个N*M 的矩形，它被划分成N*M 个边长为 1*1的小正方形区域（可以把蛋糕当成N 行 M列的矩阵）。蛋糕很快做好了，但光秃秃的蛋糕肯定不好看！所以，Sam 要在蛋糕的上表面涂抹果酱。果酱有三种，分别是红果酱、绿果酱、蓝果酱，三种果酱的编号分别为 。为了保证蛋糕的视觉效果，Admin 下达了死命令：相邻的区域严禁使用同种果酱。但 Sam 在接到这条命令之前，已经涂好了蛋糕第 K行的果酱，且无法修改。
现在 Sam 想知道：能令 Admin 满意的涂果酱方案有多少种。请输出方案数mod10^6 。若不存在满足条件的方案，请输出0 。

输入格式

输入共三行。
第一行：N,M ；
第二行： K；
第三行： M个整数，表示第K 行的方案。
字母的详细含义见题目描述，其他参见样例。

输出格式

输出仅一行，为可行的方案总数。

样例

样例输入

2 2 
1 
2 3

样例输出

3

 

数据范围与提示

对于 30% 的数据，1<=N*M<=20 ；
对于 60% 的数据， 1<=N<=1000,M<=3；
对于 100% 的数据，1<=N<=10000,M<=5。

这题正解应该是先DFS做出每一行可行解再DP但是也可以在每一行中单独判断

直接贴代码啦，注释在里面

 

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=10005;
const int mod=1e6; 
int f[N][400];//f[i][j]表示到第i行状态为j的方案数 
int x1[30],x2[30];
int sta[N];//sta[i]记下一行的第i种可行的状态 ，sta[1]为输入的第k行状态 
int poww[10];//记录3^i 
bool pd(int x);//判断单独一行的状态x是否合法 
bool check(int x,int y); //判断作为连续上下两行的x,y状态是否合法 
void init();//预处理出num种每一行可行的方案 
inline int read();//快读 
int n,m,k,a[N],num;
int main()
{
	freopen("paint.in","r",stdin);
	freopen("paint.out","w",stdout);
	scanf("%d%d%d",&n,&m,&k);
	int i,j;
	poww[0]=1;
	for(i=1;i<=m;i++)
		poww[i]=poww[i-1]*3;
	for(i=1;i<=m;i++)
	{
		a[i]=read(); 
		sta[1]=(sta[1]*3)+a[i]-1;
	} 
	if(!pd(sta[1])){printf("0\n");return 0;}
	num=1;
	init();
	if(k!=2)
	{
		for(i=1;i<=num;i++)	
			f[1][sta[i]]=1;
	}
	else
	{
		for(i=1;i<=num;i++)
			if(check(sta[i],sta[1]))
				f[1][sta[i]]=1;
	}
	int numi=0,numj=0,l;
	for(i=2;i<=n;i++)
	{
		if(i==k)numi=1;else numi=num;
		for(j=1;j<=numi;j++)//枚举i的上一行 
		{
			if(i-1==k)numj=1;else numj=num;
			for(l=1;l<=numj;l++)
			{
				if(check(sta[j],sta[l])==false)continue;
				if((i==k-1)&&check(sta[j],sta[1])==false)continue;
				f[i][sta[j]]=(f[i][sta[j]]+f[i-1][sta[l]])%mod;
			}
		}
	}
	if(n==k)numi=1;
	else numi=num;
	long long ans=0;
	for(i=1;i<=numi;i++)
	{
		ans=(ans+f[n][sta[i]])%mod;
		}	
	printf("%lld\n",ans);
	return 0;
}
inline int read()
{
	int s=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){s=s*10+ch-'0';ch=getchar();}
	return s*f;
}
bool pd(int x)
{
	int i;
	memset(x1,0,sizeof(x1));
	int l1=0;
	while(x!=0)
	{
		x1[++l1]=x%3;
		x/=3;
	}
	for(i=1;i<m;i++)
		if(x1[i]==x1[i+1])
			return false;
	return true;
}
bool check(int x,int y)
{
	int i;
	memset(x1,0,sizeof(x1));
	memset(x2,0,sizeof(x2));
	int l1=0,l2=0;
	while(x!=0)
	{
		x1[++l1]=x%3;
		x/=3;
	}
	while(y!=0)
	{
		x2[++l2]=y%3;
		y/=3;
	}
	for(i=1;i<=m;i++)
		if(x1[i]==x2[i])
			return false;
	return true;
}
void init()
{
	int i;
	for(i=0;i<poww[m];i++)
	{
		if(pd(i)==false||i==sta[1])
			continue;
		sta[++num]=i;
	} 
}