【状态压缩DP】小国王

题目来源

点我进入ACwing官网提交

题目描述

在 n×n 的棋盘上放 k 个国王，国王可攻击相邻的 8 个格子，求使它们无法互相攻击的方案总数。

输入格式
共一行，包含两个整数 n 和 k。

输出格式
共一行，表示方案总数，若不能够放置则输出0。

数据范围
1≤n≤10,
0≤k≤n2
输入样例：
3 2
输出样例：
16

题目思路

棋盘问题，求方案总数，立刻想到状态压缩DP。
棋盘问题状态压缩DP思考思路：

1. 预处理符合题意的所有相邻方案
2. 分析DP集合数组维数
3. 枚举每一种情况得出答案。

这个题目是一道很经典的棋盘问题解答。

AC代码

#include<bits/stdc++.h>
using namespace std;

#define _for(i, a, b) for (int i = (a); i < (b); ++i)
#define _rep(i, a, b) for (int i = (a); i <= (b); ++i)
#define For(i, a, b) for (int i = (a); i >= (b); --i)
#define debug(a) cout << #a << " = " << a << ENDL
#define mod(x) (x) % MOD
#define ENDL "\n"
typedef long long ll;
typedef pair<int, int> pii;
typedef vector<int> vi;

const int N = 10 + 5, M = 1 << 10, K = 100 + 5;
ll f[N][K][M];
int cnt[M], n, k;
vi path, alls[M];

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

int cal(int x) {
	int ans = 0;
	_for(i, 0, n) ans += x >> i & 1;
	return ans;
}

int main() {
#ifdef LOCAL
	freopen("data.in", "r", stdin);
#endif // LOCAL
	ios::sync_with_stdio(false);
	cin.tie(0), cout.tie(0);

	cin >> n >> k;

	_for(i, 0, 1 << n) if (check(i)) {
		path.push_back(i);
		cnt[i] = cal(i);
	}

	int sz = path.size();
	_for(i, 0, sz) _for(j, 0, sz) {
		int a = path[i], b = path[j];
		if ((a & b) == 0 && check(a | b)) alls[a].push_back(b);
	}

	f[0][0][0] = 1;
	int t;
	_rep(i, 1, n + 1) _rep(j, 0, k) _for(l, 0, sz) for (auto p : alls[t = path[l]])
		if (j >= cnt[t]) f[i][j][t] += f[i - 1][j - cnt[t]][p];

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