2018.09.10 loj#10172. 涂抹果酱(状压dp)

时间:2023-03-10 01:06:05
2018.09.10 loj#10172. 涂抹果酱(状压dp)

传送门

三进制状压感觉有点难写啊。

不过这题状态转移方程挺简单的。

就直接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;
}