Subset Sums集合 USACO-USACO阶梯-第2章.更大的挑战

最新推荐文章于 2025-03-24 08:29:27 发布

原创最新推荐文章于 2025-03-24 08:29:27 发布 · 324 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#dp #01背包

USACO习题同时被 2 个专栏收录

17 篇文章

订阅专栏

动态规划

16 篇文章

订阅专栏

该博客讨论了USACO阶梯第二章中关于Subset Sums集合的问题，涉及如何将1到N的连续整数集合划分为两个和相等的子集。通过01背包的方法，解释了解题思路，即当N为奇数时无解，否则使用动态规划求解方案数。给出了输入输出格式和样例，并提供了代码实现。

题目描述

2.2.2 Subset Sums集合

(subset.pas/c/cpp)

对于从1到N (1 <= N <= 39) 的连续整数集合，能划分成两个子集合，且保证每个集合的数字和是相等的。举个例子，如果N=3，对于{1，2，3}能划分成两个子集合，每个子集合的所有数字和是相等的：

{3} 和 {1,2}

这是唯一一种分法（交换集合位置被认为是同一种划分方案，因此不会增加划分方案总数）如果N=7，有四种方法能划分集合{1，2，3，4，5，6，7}，每一种分法的子集合各数字和是相等的:

{1,6,7} 和 {2,3,4,5} {注 1+6+7=2+3+4+5}
{2,5,7} 和 {1,3,4,6}
{3,4,7} 和 {1,2,5,6}
{1,2,4,7} 和 {3,5,6}

给出N，你的程序应该输出划分方案总数，如果不存在这样的划分方案，则输出0。程序不能预存结果直接输出（不能打表）。

格式

PROGRAM NAME: subset

INPUT FORMAT:

(file subset.in)

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

OUTPUT FORMAT:

(file subset.out)

输出划分方案总数，如果不存在则输出0。

SAMPLE INPUT

SAMPLE OUTPUT

4

解题思路：

01背包

1-N所有正整数之和: $sum=n*(n+1)/2$

等号一边的和为s1=sum/2.可以推出s1必定为偶数，否则无解

dp[i]表示和为i的方案数

则初始状态为dp[0]=1 状态转移方程为dp[j]+=dp[j-i] 终止状态为dp[sum/2]

注意：dp数组要开long long

Code:

#include<bits/stdc++.h>
#define LL long long
using namespace std;
LL n,dp[2005];
int main()
{
    scanf("%lld",&n);
    int sum = n * (n + 1);
    if(sum % 4 != 0)
    {
    	printf("0");//无解
    	return 0;
	}
	dp[0] = 1;//初始状态
	sum /= 4;//等号一边的和
	for(int i = 1;i <= n; ++i)
		for(int j = sum;j >= i; --j) dp[j] += dp[j - i];//状态转移方程
	printf("%lld\n",dp[sum] / 2);//终止状态
    return 0;
}