状态压缩dp|P1896 [SCOI2005] 互不侵犯

今天来介绍动态规划中的压缩dp算法

先来上一道题目如下P1896 [SCOI2005] 互不侵犯 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

题目描述

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

输入格式

只有一行，包含两个数 N,K。

输出格式

所得的方案数

每一次摆放国王有限制要求：

1.当前行不能有能互相攻击到的。

2.当前行的摆法不能使当前行与其他行相互攻击到。

也就是说，第一行我们可以随便摆，但是下一行也就是第二行需要满足上述两个条件，以此类推。

用暴力搜索的话时间复杂度太高了明显过不了，这里我们用压缩dp。

对于每一个格子，只有摆和不摆两个选项。假如用十进制的5表示二进制的0101，所代表的摆法是

绿色是不摆，黄色是摆。

定义dp[i][j][k]表示前i行，已经摆了j个国王，且第i行状态为k（比如上图）有多少种摆法。

那么题目问的就是前n行，已经摆了m个国王，所有合法状态的和

先给出状态转移方程：dp[i][j][a] += dp[i-1][j-c][x];

假设我们现在要摆第i行了，我们希望第1~i行总共有j个国王，且第i行的状态（摆法）为a。

这种情况的总数为dp[i][j][a]。

毫无疑问，第i行的摆法是受前i-1行摆法的限制的。

第一个限制是数量，我们希望1~n行一共有j个国王，那么前i-1行国王数量应该小于等于j。

第二个限制是不能使得第i-1行和第i行相互攻击。

假如存在满足以上两种的，在前i-1行的摆法，我们就可以通过这些摆法，实现我们之前所希望的dp[i][j][a]

现在我们确定了前i行的国王数量，也确定了第i行的状态。假设dp[i-1][x1][x2]=x3,dp[i-1][y1][y2]=y3

假设dp[i-1][x1][x2]=x3,dp[i-1][y1][y2]=y3【这样的摆法】可以使得我们希望的第i行摆法成立。

那么dp[i][j][a]=所有【这样的摆法】的累加。

#include<bits/stdc++.h>
using namespace std;
long long dp[11][110][1 << 10];
vector<int> state;
int n, m;
int cnt[1 << 10];
vector<vector<int>> ma;

bool check(int x) {
    for(int i = 0; i < n; i++) {
        if((x >> i & 1) && (x >> (i + 1) & 1)) return false;
    }
    return true;
}

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

int main() {
    cin >> n >> m;
    dp[0][0][0] = 1;
    
    for(int i = 0; i < (1 << n); i++) {
        if(check(i)) {
            state.push_back(i);
            cnt[i] = count(i);
        }
    }

    ma.resize(state.size());
    for(int i = 0; i < state.size(); i++) {
        for(int j = 0; j < state.size(); j++) {
            int a = state[i];
            int b = state[j];
            if((a & b) == 0 && check(a | b)) {
                ma[i].push_back(j);
            }
        }
    }
    
    for(int i = 1; i <= n; i++) {
        for(int j = 0; j <= m; j++) {
            for(int a = 0; a < state.size(); a++) {
                for(int x : ma[a]) {
                    int c = cnt[state[a]];
                    if(j >= c) dp[i][j][a] += dp[i-1][j-c][x];
                }
            }
        }
    }

    long long result = 0;
    for(int i = 0; i < state.size(); i++) {
        result += dp[n][m][i];
    }
    
    cout << result << endl;
    return 0;
}