集合划分差最小问题(01背包)

集合划分差最小问题(01背包)

一个序列分为两个集合，使得两个集合数的和差值最小。

考虑背包，容量为$\frac{sum}{2}$，转化为判定问题。

#include<bits/stdc++.h>
using namespace std;
const int N=5e3+5;
int n,s;
int a[N];
bool dp[N];
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]),s+=a[i];
    int m = s>>1;
    dp[0] = true;
    for(int i=1;i<=n;i++)
    	for(int j=m;j>=a[i];j--)
    		dp[j]|=dp[j-a[i]];
    int ans = 0;
    for(int i=m;i;i--){
    	if(dp[i]){
    		ans=s-(i<<1);
    		break;
    	}
    }
    printf("%d\n",ans);
    return 0;
}

如果要输出方案，就再开一个$pre$数组记录使用的数，然后dfs输出路径。