洛谷P1896 互不侵犯 状态压缩dp

题目描述

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

注：数据有加强（2018/4/25）
输入输出格式
输入格式：

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

输出格式：

所得的方案数

输入输出样例
输入样例#1：

3 2

输出样例#1：

16

经过前两题的熏陶，关键容易多了，这道题有行数，皇帝的个数，所以加上一维状态，就是3维的数组，dp【i】【j】【k】 表示第i行是j状态的情况下，前i行用的皇帝的个数是k的种类。
can数组表示一行可行的状态，need表示可行的状态下所需要的皇帝数。need也就是存没种状态下有几个1就是用了几个皇帝。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int n, peo;
ll dp[11][1500][90];
int need[1000];
int can[1000];
int main()
{
    cin >> n >> peo;   //我用peo表示给的人数了，用k不习惯。
    ll mm = (1 << n) -1;
    int num = 0;
    for (int i = 0; i <= mm; i++)
    {
        if(!((i << 1) & i))  //这种状态可行，下边求下有几个1就ok。
        {
            can[++num] = i;
            int j = i;
            while(j)
            {
                need[num] += j%2;
                j/=2;
            }
        }
    }
    for (int i = 1; i <= num; i++)
        if(need[i] <= peo) dp[1][i][need[i]] = 1; //可行状态需要人数小于所拥有的人。
	//我们得预处理第一行，因为不像前两题，dp【0】【0】 = 1 这个题得先找出一些1。
	
    for (int i = 2; i <= n; i++)   //层数
    for (int j = 1; j <= num; j++) //这一层的状态
    for (int k = 1; k <= num; k++) //上一层的状态
    {
        if(can[j] & can[k]) continue;
        if(can[j] & (can[k] << 1)) continue;
        if((can[j] << 1) & can[k]) continue;  //这三行是判断上一层第k种状态，和这一次呢个第j种状态是题中的“相邻”
        for (int t = 1; t <= peo; t++)  //枚举节制上一层所用人数， 再复习一次
        {    //dp【i】【j】【k】 表示第i行是j状态的情况下，前i行用的皇帝的个数是k的种类。
            if(need[j] + t > peo) continue;  //这一层j状态需要的人数是need【j】
            dp[i][j][need[j]+t] += dp[i-1][k][t];  
        }
    }

    ll sum = 0;
    for (int i = 1; i <= n; i++)
    for (int j = 1; j <= num; j++)
    sum+=dp[i][j][peo]; //用peo个皇帝的各种情况

    cout << sum << endl;
    return 0;
}