P1896 [SCOI2005]互不侵犯(基础状压DP）

P1896 [SCOI2005]互不侵犯

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

输入格式
只有一行，包含两个数N，K （ 1 <=N <=9, 0 <= K <= N * N）

输出格式
所得的方案数

输入输出样例
输入
3 2
输出
16
思路：
　　　因为每个国王只影响自己周围的一格，一行(一列)都可以放好几个国王， ，所以，我们考虑dp，把每一行做成一个整体。dp方程:f[i][j][S]=∑f[i-1][k][T]
为什么这样写呢？我先阐述一下这个转移方程的意思：f[i][j][S]代表: 当第i行状态为j时，已经放置了S个国王的总方案数。 因为每个国王都是只影响上下两行的(列的情况先丢一边)，不过我们只管上一行(当到当前行的下一行时，这行的情况自然会被匹配)，而对于k这个状态，只要它不能更j这个状态冲突，就可以加上f[i-1][k][T]。T跟S的意思大致相同，代表放置的国王数。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define N 11
using namespace std;
int  pre[1<<N],num[1<<N],cnt;
long long dp[N][1<<N][N*N];
int getnum(int x)
{
    int sum=0;
    while (x) sum+=(x&1),x>>=1;
    num[cnt]=sum;
    return sum;
}
int main()
{
    int n,k;
    cin>>n>>k;
    for(int i=0;i<(1<<n);i++){//状态从0开始！！！表示第一行没有国王的情况
        if(!(i&(i<<1))){//如果 i 的二进制有连续两个数都为 1 那么i&i-1一定不为0，
        //反之若不存在给情况
            pre[++cnt]=i;
            dp[1][cnt][getnum(i)]=1;
        }
    }
    int x,y;
    for(int i=2;i<=n;i++){
        for(int j=1;j<=cnt;j++){
            x=pre[j];
            for(int p=1;p<=cnt;p++){
                y=pre[p];
                if((x&y)||(x&(y>>1))||(x&(y<<1))) continue;
                for(int s=0;s<=k;s++) {
                    dp[i][j][num[j]+s]+=dp[i-1][p][s];
                }
            }
        }
    }
    long long sum=0;
    for(int i=1;i<=cnt;i++){
        sum+=dp[n][i][k];
    }
    cout<<sum<<endl;
}