[USACO2.2]集合 Subset Sums解析

题目链接

题意

给出 1∼n 的连续整数集合，求划分成两个和相等的子集合的方案数，如果不能划分，则输出0。

题目分析

显然，当${\sum }_{i=1}^{n}i$ 为奇数时，一定不能划分成功，为偶数时，一定能划分。

论证 ${\sum }_{i=1}^{n}i$ 为偶数时，一定能划分成功。
${\sum }_{i=1}^{n}i$ = n*(n+1)/2 为偶数，则：

• n = 4m （m为正整数）
• n+1 = 4m （m为正整数）

当n = 4m时，可以按首尾一组，刚好可以分成偶数组，即1与n，2与n-1，…,每组的和相同，偶数组分成两个组数相同的子集，其和必定相等。
当n+1 = 4m时，n = 4m -1 = 4m + 3，将1 2 3 分成1 2一组和3一组，剩余数字按上述原理仍然可以分成和相等的两个子集。

当${\sum }_{i=1}^{n}i$ 为偶数时，问题为经典的【01背包问题】

定义dp[i][j]状态为：前 i 个数字，和为 j 的方案数。

状态转移：

• j < i (不能选i) ：dp[i][j] = dp[i-1][j]
• j == i ： dp[i][j] = dp[i-1][j] + 1 (不选i的方案数 + 选i的方案数)
• j > i : dp[i][j] = dp[i-1][j] + dp[i-1][j-i] （不选i的方案数+选i的方案数）

其中，后两种情况可以并为一种，只需要将dp[][0] = 1 即可。
最后将01背包进行空间优化。
【注】对于集合{1,2,3}而言，划分成{1,2} 和 {3}这是一种划分方案，因此最终的答案为dp[sum/2] / 2。
代码如下：

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int maxn = 40;
int n;
int dp[maxn];
int cal(int m){
    dp[0] = 1;
    for(int i = 1;i<=n;i++){
        for(int j = m;j>=i;j--){
            dp[j] = dp[j]+dp[j-i];
        }
    }
    return dp[m] / 2;
}
signed main(){
    cin>>n;
    int sum = (n+1)*n / 2;
    if(sum%2 == 1) cout<<0<<endl;
    else cout<<cal(sum/2)<<endl;
    return 0;
    
}