P1466 [USACO2.2] 集合 Subset Sums

题目链接

Luogu P1466

题面

题目描述

对于从 $1\sim n$ 的连续整数集合，能划分成两个子集合，且保证每个集合的数字和是相等的。举个例子，如果 $n=3$，对于 { 1 , 2 , 3 } \{1,2,3\} {1,2,3} 能划分成两个子集合，每个子集合的所有数字和是相等的：
{ 3 } \{3\} {3} 和 { 1 , 2 } \{1,2\} {1,2} 是唯一一种分法（交换集合位置被认为是同一种划分方案，因此不会增加划分方案总数）
如果 $n=7$，有四种方法能划分集合 { 1 , 2 , 3 , 4 , 5 , 6 , 7 } \{1,2,3,4,5,6,7 \} {1,2,3,4,5,6,7}，每一种分法的子集合各数字和是相等的:
{ 1 , 6 , 7 } \{1,6,7\} {1,6,7} 和 { 2 , 3 , 4 , 5 } \{2,3,4,5\} {2,3,4,5}
{ 2 , 5 , 7 } \{2,5,7\} {2,5,7} 和 { 1 , 3 , 4 , 6 } \{1,3,4,6\} {1,3,4,6}
{ 3 , 4 , 7 } \{3,4,7\} {3,4,7} 和 { 1 , 2 , 5 , 6 } \{1,2,5,6\} {1,2,5,6}
{ 1 , 2 , 4 , 7 } \{1,2,4,7\} {1,2,4,7} 和 { 3 , 5 , 6 } \{3,5,6\} {3,5,6}
给出 $n$，你的程序应该输出划分方案总数。

输入格式

输入文件只有一行，且只有一个整数 $n$

输出格式

输出划分方案总数。

样例 #1

样例输入 #1

7

样例输出 #1

4

提示

【数据范围】
对于 $100\%$ 的数据，$1\le n\le 39$。
翻译来自NOCOW
USACO 2.2

题意

有一个正整数 $n$ ，现有一个从 $1\sim n$ 的连续整数集合，将其划分为两个集合，使两个集合的元素和相等，求有多少中划分方案（交换集合位置被认为是同一种划分方案，因此不会增加划分方案总数）

思路

先特判一手，若 $1\sim n$ 的连续整数集合的元素和为奇数，则此集合是无论如何都不能按照上述方式划分的，输出 $0$
将集合中的元素视为阶段，用 $01$ 背包枚举一些元素放入前一个集合，使前一个集合的元素和与后一个集合的元素和相等，对前一个集合的元素和进行分类讨论
则 $dp[i]$ 代表前一个集合的元素和为 $i$ 时的方案数
若当前的元素 $i$ 不放入前一个集合里，则 $dp$ 数组不必更新，跳过
若当前的元素 $i$ 放入前一个集合里，则 $dp[i]+=dp[i-j]$
并且还要对 $dp$ 数组初始化

$dp[0]=1$

具体见代码

#include<bits/stdc++.h>
using namespace std;
#define int long long
int dp[(1+40)*40/2]={1LL},sum,n;
signed main(){
	scanf("%lld",&n);
	sum=(1+n)*n/2;//元素总和
	if(sum%2){//集合内的元素和为奇数，无法划分
		printf("0");
		return 0;
	}
	for(int i=1;i<=n;i++){
		for(int j=sum/2;j>=i;j--){//只需填满前一个集合，后一个集合也就出来了，所以用sum/2
			dp[j]+=dp[j-i];//01背包——状态转移方程
		}
	}
	printf("%lld",dp[sum/2]/2);//输出时要除以2，因为有两个集合（交换集合位置被认为是同一种划分方案，因此不会增加划分方案总数）
	return 0;
}