hdu_4529_郑厂长系列故事——N骑士问题(状压DP)

题目连接：http://acm.hdu.edu.cn/showproblem.php?pid=4529

题意：中文，不解释

题解：状压DP，dp[i][j][k][s]表示第i行当前用了j个骑士，i-1行的压缩状态为k,i行的压缩状态为j，然后用滚动数组优化了一下，注意如果不预处理不可存放位置会超时

#include<cstdio>
#include<cstring>
#define N (1<<8)
#define FFC(i,a,b) for(int i=a;i<=b;++i)
    
int T,n,dp[2][23][N][N],g[N],num[N],f1[N][N],f2[N][N],cur,ans;

void init(){//预处理不可放的位置，和每一个压缩状态的骑士数目
    FFC(i,0,N-1){
        FFC(j,0,7)if(i&(1<<j))num[i]++;
        FFC(j,0,N-1){
            if(((i>>2)&j)||((j>>2)&i))f1[i][j]=1;
            if(((i>>1)&j)||((j>>1)&i))f2[i][j]=1;
        }
    }
}
int fuck(){
    memset(dp,0,sizeof(dp));
    cur=0,ans=0,dp[0][0][0][0]=1;
    FFC(i,0,7){
        FFC(j,0,n)FFC(p,0,N-1)FFC(q,0,N-1){
            if (dp[cur][j][p][q]==0)continue;
            FFC(z,0,N-1)
                if (((z&g[i+1])!=z)||num[z]+j>n||(i>=1&&f1[q][z])||(i>=2&&f2[p][z]))continue;
                else dp[cur^1][num[z]+j][q][z]+=dp[cur][j][p][q];
        }
        memset(dp[cur],0,sizeof(dp[cur])),cur^=1;
    }
    FFC(i,0,N-1)FFC(j,0,N-1)ans+=dp[cur][n][i][j];
    return ans;
}

int main() {
    init();char str[N];
    scanf("%d",&T);
    while (T--){
        scanf("%d",&n);    
        memset(g,0,sizeof(g));
        FFC(i,1,8){
            scanf("%s",str);
            FFC(j,0,7){
                g[i]<<=1;
                if(str[j]=='.')g[i]|=1;
            }
        }
        printf("%d
",fuck());
    }
    return 0;
}