P1896 [SCOI2005]互不侵犯-动态规划，状压dp

在N×N的棋盘里面放K个国王，使他们互不攻击，共有多少种摆放方案。国王能攻击到它上下左右，以及左上左下右上右下八个方向上附近的各一个格子，共8个格子。

https://www.luogu.org/problemnew/show/P1896

 

#include<iostream>
#include<cstdio>
#include<cstring> 
#define ll long long
using namespace std;
ll n,k,cnt,ans;
bool a[1<<9];
ll s[1<<9],f[10][1<<9][91];
int main()
{
    scanf("%lld%lld",&n,&k);
    memset(f,0,sizeof(f));
    for(int x=0;x<(1<<n);x++){
        if(!(x&(x>>1))&&!(x&(x<<1)))a[x]=1;
        int w=x;
        while(w){
            if(w%2) s[x]++;
            w/=2;
        }
        if(a[x]) f[1][x][s[x]]=1;
    }
    for(int i=2;i<=n;i++){
        for(int q=0;q<(1<<n);q++){
            if(a[q]){
                for(int p=0;p<(1<<n);p++){
                    if(!(q&p)&&!(q&(p<<1))&&!(q&(p>>1))&&a[p]){
                        for(int j=0;j+s[p]<=k;j++){
                            f[i][p][j+s[p]]+=f[i-1][q][j];
                        }
                    }
                }
            }
        } 
    }
    for(int i=0;i<(1<<n);i++){
        ans+=f[n][i][k];
    }
    printf("%lld",ans);
    return 0;
}