一、题目描述
二、算法分析说明
本博客是根据 https://www.cnblogs.com/kangjianwei101/p/5332451.html 改写的。
设集合中全部元素(1 ~ N)的和为 t ,当 t 为奇数时,因为奇数 + 奇数 = 偶数,偶数 + 偶数 = 偶数,显然不可能将总和为奇数的元素分成两组和相等的元素。此时直接输出 0 。
当 t 为偶数时,分成的两个子集的元素和相等,均为 t / 2 。
设 dp[i][j] 为总和不超过 j 的情况下,从元素 1 到 i 中选择若干个数放入子集的方案数。注意,i 是正在考虑是否放进集合的数,不代表是否选择放入 i 。
①当 j < i 时,i 显然不能放进子集,否则总和超过限制。这时只能照搬决策过的前 i - 1 个元素的方案总数,所以
dp[i][j] = dp[i - 1][j];
②当 j >= i 时,i 可以放入子集。但是无论是否放入都可能可以作为可行方案。所以,要将放入和不放入的方案数(两者均可以为零)都加起来。i 不放入,方案数自然与已经决策过的 1 ~ i - 1 元素的选择范围内的方案数相等。如果 i 要放入,则剩余 j - i 的和未决策,这部分需要查找前面已经计算过的结果获得答案。循环是正序进行的,所以从前 i - 1 个数中选择的可选方案数肯定已经决策过。
初始条件一共有:
dp[0][j] = 0, j = 0, 1, 2, …, t;(注意,这里已经进行了 t >>= 1 操作)
这个条件用于决策第一个数时参考,因为这之前没有发现任何可行方案,所以之前的累积自然只能为零。
dp[1][0] = 1, dp[1][1] = 1;
这两个条件代表:和为 0 时,1 不能放入,只有这一种方案;和为 1 时,1 可以放入,如放入 1 ,剩下未决策的和为 0 ,已经发现的方案数为零;如不放入 1 则已经累积的结果 dp[0][1] = 0 ,所以加起来也只有这一种方案。
递推完毕后,直接输出结果。
三、AC 代码(1 ms)
#include<cstdio>
#include<vector>
using namespace std;
#pragma warning(disable:4996)
vector<unsigned> v; unsigned n, t, dp[40][781];
int main() {
scanf("%u", &n); t = n * (n + 1) / 2;
if (t & 1)puts("0");
else {
t >>= 1, dp[1][0] = 1, dp[1][1] = 1;
for (unsigned i = 2; i <= n; ++i) {
for (unsigned j = 0; j <= t; ++j) {
if (j < i)dp[i][j] = dp[i - 1][j];
else dp[i][j] = dp[i - 1][j] + dp[i - 1][j - i];
}
}
printf("%u\n", dp[n][t] / 2);
}
return 0;
}