洛谷P1896 [SCOI2005]互不侵犯King

链接

　　https://www.luogu.org/problem/show?pid=2051

题解

　　经典状压DP题，f[i][j][k]表示已经做了i行，放了j个国王，当前行的状态是k。
　　转移的时候直接枚举上一行和当前行的状态转移就行了。

代码

//状压dp
#include <cstdio>
#include <algorithm>
#define maxn 10
#define ll long long
using namespace std;
ll N, K, s[1<<maxn], f[maxn][100][1<<maxn];
void dfs(ll pos, ll x)
{
    if(pos==N){s[++s[0]]=x;return;}
    dfs(pos+1,x);
    if((x&(1<<pos-1))==0)dfs(pos+1,x|(1<<pos));
}
bool check(ll x, ll y)
{
    if(x&y)return false;
    for(ll t=x|y;t;t>>=1)if((t&1) and (t&2))return false;
    return true;
}
ll cnt(ll x)
{
    ll cnt=0;
    for(;x;x>>=1)cnt+=x&1;
    return cnt;
}
int main()
{
    ll i, j, f1, f2, ans=0;
    scanf("%lld%lld",&N,&K);
    dfs(0,0);
    f[0][0][1]=1;
    for(i=1;i<=N;i++)
    {
        for(j=0;j<=K;j++)
        {
            for(f1=1;f1<=s[0];f1++)
            {
                for(f2=1;f2<=s[0];f2++)
                {
                    if(check(s[f1],s[f2])==false or cnt(s[f1]|s[f2])>j)continue;
                    f[i][j][f2]+=f[i-1][j-cnt(s[f2])][f1];
                }
            }
        }
    }
    for(i=1;i<=s[0];i++)ans+=f[N][K][i];
    printf("%lld\n",ans);
    return 0;
}