状态压缩dp-小国王

小国王

思路：
fijk：前i行已经放好，放了j个，第i行状态是k。dp前预处理所有合法状态，并且处理状态之间的转移关系，然后dp直接转移即可。
状态转移方程：

f[i][j][s] += f[i-1][j-c][b];

代码：

#include <iostream>
#include <vector>
using namespace std;
const int N= 12,M = 1<<10,K = 110;
int n,m;
vector<int> state;
int id[M],cnt[M];
vector<int> head[M];
long long  f[N][K][M];
bool check(int s){
    for(int i = 0;i<n;i++){
        if((s>>i& 1 )&&(s>>i+1 &1))
            return false;
    }
    return true;
}

int count(int s){
    int res = 0;
    for(int i = 0;i<n;i++){
        res+= s >> i & 1;
    }
    return res;
}

int main() {
    cin>>n>>m;
    for(int i = 0;i<1<<n;i++){
        if(check(i)){
            state.push_back(i);
            id[i] = state.size()-1;
            cnt[i] = count(i);
        }
    }
    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&& check(a|b))
                head[i].push_back(j);
        }
    }
    f[0][0][0] = 1;
    for(int i = 1;i<=n+1;i++){
        for(int j = 0;j<=m;j++){
            for(int s = 0;s<state.size();s++){
                for(auto b: head[s]){
                    int c = cnt[state[s]];
                    if (j>=c){
                        f[i][j][s]  += f[i-1][j-c][b];
                    }
                }
            }
        }
    }

    cout<<f[n+1][m][0];
    return 0;
}