将一堆正整数分为2组,要求2组的和相差最小。
例如:1 2 3 4 5,将1 2 4分为1组,3 5分为1组,两组和相差1,是所有方案中相差最少的。
收起
输入
第1行:一个数N,N为正整数的数量。 第2 - N+1行,N个正整数。 (N <= 100, 所有正整数的和 <= 10000)
输出
输出这个最小差
输入样例
5 1 2 3 4 5
输出样例
1
一开始想到的是每个数有加和不加两种可能,然后套了个dfs。结果超时。
超时代码如下:
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <math.h>
using namespace std;
const int maxn=105;
int n;
int a[maxn];
int re=0x3f3f3f3f;
int vis[maxn];
int Sum=0;
void dfs (int loc,int sum)
{
re=min(abs(Sum-2*sum),re);
for (int i=loc;i<n;i++)
{
if(!vis[i])
{
vis[i]=1;
dfs(i,sum+a[i]);
vis[i]=0;
}
}
}
int main()
{
scanf("%d",&n);
memset (vis,0,sizeof(vis));
for (int i=0;i<n;i++)
{
scanf("%d",&a[i]);
Sum+=a[i];
}
dfs(0,0);
printf("%d\n",re);
return 0;
}
之后看了讨论区,发现是01背包。
将总和/2之后套用0-1背包来做。
代码如下:
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <math.h>
using namespace std;
const int maxn=105;
int n;
int a[maxn];
int re=0x3f3f3f3f;
int vis[maxn];
int Sum=0;
void dfs (int loc,int sum)
{
re=min(abs(Sum-2*sum),re);
for (int i=loc;i<n;i++)
{
if(!vis[i])
{
vis[i]=1;
dfs(i,sum+a[i]);
vis[i]=0;
}
}
}
int main()
{
scanf("%d",&n);
memset (vis,0,sizeof(vis));
for (int i=0;i<n;i++)
{
scanf("%d",&a[i]);
Sum+=a[i];
}
dfs(0,0);
printf("%d\n",re);
return 0;
}