51nod - 1007 正整数分组 dp水题 练习

将一堆正整数分为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

解题思路：

转换成背包问题

dp[i][j]表示从前i个数中选择若干个数，总和不超过j的时候能得到的最大的值。

即在前n个数中求得总和不超过sum/2的数值

答案等于sum-2*dp[n][sum/2]

#include <deque>
#include <vector>
#include <queue>
#include <string>
#include <cstring>
#include <map>
#include <stack>
#include <set> 
#include <iostream>
#include <cstdio>
#include <fstream>
#include <algorithm>
#include <cmath> 
#include <limits>
using namespace std;

typedef long long LL;
LL a[10000];
LL dp[110][1000];
LL i,j;
int main()
{
    int n;
    cin >> n;
    LL sum=0;;
    for(i=1; i<=n; i++)
    {
        cin >> a[i];
        sum += a[i];
    }
    memset(dp,0,sizeof(dp));
    for(int i=1; i<=n; i++)
        for(int j=0; j<=sum/2; j++)
            if(j<a[i])
                dp[i][j]=dp[i-1][j];
            else
                dp[i][j]=max(dp[i-1][j],dp[i-1][j-a[i]]+a[i]);
    cout<<sum-2*dp[n][sum/2]<<endl;
}

优化空间 一位数组

#include <deque>
#include <vector>
#include <queue>
#include <string>
#include <cstring>
#include <map>
#include <stack>
#include <set> 
#include <iostream>
#include <cstdio>
#include <fstream>
#include <algorithm>
#include <cmath> 
#include <limits>
using namespace std;

typedef long long LL;
LL a[10000];
LL dp[10000];
LL i,j;
int main()
{
    int n;
    cin >> n;
    LL sum=0;;
    for(i=1; i<=n; i++)
    {
        cin >> a[i];
        sum += a[i];
    }
    memset(dp,0,sizeof(dp));
    for(i=1; i<=n; i++)
        for(j=sum/2; j>=a[i]; j--)
            dp[j]=max(dp[j],dp[j-a[i]]+a[i]);
    cout << sum-2*dp[sum/2] << endl;
}