小国王（状态压缩dp）

 

#include <iostream>
#include <cstring>
#include <cstring>
#include <vector>

using namespace std;

typedef long long LL;

const int N = 12,M = 1 << 10,K = 110;

int n,m;
vector<int > state;//存储每一种合法状态
int cnt[M];//存储每个状态中1的数量
vector<int > head[M];//存储对于每一种合法状态可以转移到的合法状态
LL f[N][K][M];

bool cheak(int state) //检查是否有相邻的1
{
    for(int i = 0;i < n;i ++)
        if((state >> i & 1) && (state >> (i + 1 )& 1))
            return false;
            
    return true;
}

int count(int state)//计算有几个1
{
    int res = 0;
    for(int i = 0;i < n;i ++) res += state >> i & 1;
    return res;
    
}
int main()
{
    cin >> n >> m;
    
    for(int i = 0;i < 1 << n;i ++) //遍历第一行的所有状态，找出所有合法状态 也可以说找出单独每一行的所有合法状态
        if(cheak(i))
        {
            state.push_back(i);
            cnt[i] = count(i);//记录每一种合法状态的1的数量及放了几个国王
        }
        
    for(int i = 0;i < state.size();i ++)//找出第一行每一种状态所能转移到的所有合法状态
        for(int j = 0;j < state.size();j ++)
        {
            int a = state[i],b = state[j];
            if((a & b) == 0 && cheak(a | b)) //a & b == 0 不存在上下相邻的两个1 cheak（a | b）不存在纵坐标相邻的两个1
                head[i].push_back(j);//存储合法状态
        }
        
//dp
    f[0][0][0] =  1; //前0行已经摆了0个国王的方案只有0 这1种
    for(int i = 1;i <= n + 1;i ++)// 遍历每一行
        for(int j = 0;j <= m;j ++)//国王数量
            for(int a = 0;a < state.size();a ++)
                for(int b : head[a])
                {
                    int c = cnt[state[a]];
                    if(j >= c)
                        f[i][j][a] += f[i - 1][j - c][b];
                }
                
    cout << f[n + 1][m][0] << endl;
    //就是我们假设存在第n+1行，但是第n+1行完全没有摆
    //那这种情况里面的所有方案其实就是等于这个这个我们只有n行的一个所有方案
    //那这样枚举n+1的一个好处是我们最后不需要再循环枚举最后一行的状态了
    //就是我们这个f[n+1][m][0]已经在这个循环里面被循环算出来了
    //所以可以少一层循环
    return 0;
    
}