传送门
三进制状压感觉有点难写啊。
不过这题状态转移方程挺简单的。
就直接f[i][j]表示前i行第i行状态为j时的选法总数,分情况转移就行了。
代码:
#include<bits/stdc++.h>
#define ll long long
#define mod 1000000
#define N 10005
using namespace std;
int n,m,K,ban,ans=0,sta[1005],tot=0,stat,f[N][1005],bit[6],pos;
inline bool check(int x){
int tmp=0x3f;
for(int i=1;i<=m;++i){
if(tmp==x%3)return false;
tmp=x%3,x/=3;
}
return true;
}
inline int read(){
int ans=0;
char ch=getchar();
while(!isdigit(ch))ch=getchar();
while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
return ans;
}
inline bool judge(int a,int b){
for(int i=1;i<=m;++i){
if(a%3==b%3)return false;
a/=3,b/=3;
}
return true;
}
int main(){
n=read(),m=read(),K=read(),stat=1;
for(int i=1;i<=m;++i)stat*=3;
for(int i=0;i<stat;++i)if(check(i))sta[++tot]=i;
for(int i=1;i<=m;++i)ban=ban*3+read()-1;
for(int i=1;i<=tot;++i)if(ban==sta[i]){pos=i;break;}
if(!pos){puts("0");return 0;}
for(int i=1;i<=n;++i){
if(i==K){
if(i==1)f[i][pos]=1;
else for(int j=1;j<=tot;++j)if(judge(sta[pos],sta[j]))
(f[i][pos]+=f[i-1][j])%=mod;
}
else for(int j=1;j<=tot;++j){
if(i==1)f[i][j]=1;
else for(int k=1;k<=tot;++k)if(judge(sta[j],sta[k]))
(f[i][j]+=f[i-1][k])%=mod;
}
}
for(int i=1;i<=tot;++i)(ans+=f[n][i])%=mod;
cout<<ans;
return 0;
}