LA 4794 Sharing Chocolate 状压dp--个人的理解+带注释代码

原题： https://vjudge.net/problem/UVALive-4794

题意：  给你一个a*b的巧克力块，问能不能分为面积分别为 a1，a2……an 的巧克力块（n<=15)，每次操作只能横着切或者竖着切，不能切一半不切一半。

 

首先看数据量，n<=15,这么特殊的数据量很容易让人想到状态压缩dp。

我们将需要得到的小巧克力面积用一个二进制保存起来，同时定义一个num数组来对应二进制的值，这是状态压缩dp的基本操作。

不知道枚举子集和不太了解状态压缩dp的，我之前写过一篇博客https://blog.csdn.net/weixin_43191865/article/details/89217744

那么我们如何确定dp数组呢，只有一个状态是不够的。或许我们可以从其他方向入手

我们在对巧克力操作的时候，能影响到操作的是什么，只有巧克力的长和宽。

知道巧克力的长和宽，我就可以知道所有它能得到的巧克力，因此，我们的状态或许还需要知道加上长和宽

那么我们可以设一个三维数组 dp[a][b][s]  a: 长，b：宽， s：二进制    

好了下一个问题来了，我们这个二进制究竟要表示什么

很容易想到，s表示： a*b的巧克力所能得到的所有小巧克力块。

然而三维的dp数组运算代价是很恐怖的，我们需要简化它

由于题目的小巧克力块加起来等于a*b，因此我只需要知道a，b的其中一个以及总面积，就可以运算另外一个了。

因此我们再引入一个数组 sum， 下标是二进制的状态，保存的则是该状态下的巧克力面积   

b=sum【s】 /a

因此我们可以这么保存dp：

dp[i][s]  :  i=min(a,b)  s:二进制 ， dp为bool型，1表示a*b的巧克力可以分离出s的小巧克力

 

接下来考虑状态转移方程

当dp[i][s]为1时，当且仅当存在一个大小为1~i的j，使得

dp[j][s0] 和 dp[j][s-s0]  都为真，其中s0 为s的子集合。

这个j的意思就是：将巧克力切成其中一条边长为j的小巧克力。

可能又有问题了，我怎么知道j所对应的s0呢，我也不知道，那么就遍历吧。将s的所有子状态s0都遍历一次，分别判断

代码： （因为yes no大小写的原因找了好久bug）

 

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1 << 17;
int dp[maxn][107], sum[maxn], num[108];
// dp: 就是那个dp   sum: 状态为s的时候的面积总和（方便求j）    num：小巧克力块
int n;
int check(int num) { //假如s中只剩下一块巧克力了，那么就说明这种方法可以，返回1
	return num == 0 ? 0 : check(num >> 1) + (num & 1);
}
bool DP(int s, int x) {
	if (dp[s][x] >= 0) return dp[s][x]; //记忆话搜索的操作
	if (check(s) == 1) return dp[s][x] = 1;
	int y = sum[s] / x;
	for (int s0 = (s - 1)&s; s0; s0 = (s0 - 1) & s) { //遍历s的子集的
		int s1 = s - s0;
		// 不需要遍历边了，知道面积和一条边就能知道另外一条边
		if (sum[s0] % x == 0 && DP(s0, min(x, sum[s0] / x)) && DP(s1, min(x, sum[s1] / x)))
			return dp[s][x] = 1;
		// 上面下面对应横竖切巧克力
		if (sum[s0] % y == 0 && DP(s0, min(y, sum[s0] / y)) && DP(s1, min(y, sum[s1] / y)))
			return dp[s][x] = 1;
	}
	return dp[s][x] = 0;
}
int main() {
	int cnt = 0;
	while (cin >> n, n) {
		memset(dp, -1, sizeof(dp)); // -1 没搜索过   1： 可以  0： 不可以
		memset(sum, 0, sizeof(sum));
		int x, y; cin >> x >> y;
		int ss = 0;
		for (int i = 0; i < n; i++) {
			scanf("%d", num + i);
			ss += num[i];
		}
		for (int s = 0; s < (1 << n); s++) {   // 状态所对应的面积和存入sum数组中
			for (int i = 0; i < n; i++) if ((s >> i) & 1 != 0) sum[s] += num[i];
		}
		int res;
		if (sum[(1 << n) - 1] != x * y || sum[(1 << n) - 1] % x != 0) { //假如一开始给的小巧克力面积就不等于大巧克力，那就不用算了
			res = 0;
		}
		else {
			res = DP((1 << n) - 1, min(x, y)); 
		}
		cout << "Case " << ++cnt << ": ";
		if (res == 1) {
			cout << "Yes" << endl;
		}
		else cout << "No" << endl;
	}
	return 0;
}
/*
4
3 4
6 3 2 1
2
2 3
1 5
0
*/