BZOJ 1087 [SCOI2005]互不侵犯King

暴力状压dp。。。

好久不写状压dp，先水一个~

#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <algorithm>

#define N 1000

using namespace std;

int num,n,m;
int zt[N],gs[N];
long long dp[11][1<<11][90];
bool map[N][N];

inline bool check(int x,int y)
{
    if(x&(y<<1)) return false;
    if(x&(y>>1)) return false;
    if(x&y) return false;
    return true;
}

inline bool judge(int x)
{
    if(x&(x<<1)) return false;
    if(x&(x>>1)) return false;
    return true;
}

inline int getnum(int x)
{
    int res=0;
    while(x)
    {
        if(x&1) res++;
        x>>=1;
    }
    return res;
}

inline void prev()
{
    int sum=(1<<n)-1;
    for(int i=0;i<=sum;i++)
        if(judge(i))
        {
            zt[++num]=i;
            gs[num]=getnum(i);
        }
    for(int i=1;i<=num;i++)
        for(int j=i;j<=num;j++)    
            if(check(zt[i],zt[j]))
                map[j][i]=map[i][j]=true;
}

inline void go()
{
    dp[0][1][0]=1LL;
    for(int i=0;i<n;i++)
        for(int j=1;j<=num;j++)
            for(int k=0;k<=m;k++)
                if(dp[i][j][k])
                    for(int p=1;p<=num;p++)
                        if(map[j][p]&&k+gs[p]<=m)
                            dp[i+1][p][k+gs[p]]+=dp[i][j][k];
    long long sum=0;
    for(int i=1;i<=num;i++)
        sum+=dp[n][i][m];
    printf("%lld\n",sum);
}

int main()
{
    scanf("%d%d",&n,&m);
    prev();
    go();
    return 0;
}