题目描述
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;
}
}