D: 小李打怪兽I

D: 小李打怪兽I

时间限制: 1 Sec 内存限制: 128 MB

题目描述

小李对故乡的思念全部化作了对雾霾天气的怨念，这引起了掌控雾霾的邪神的极大不满，邪神派去了一只小怪兽去对付小李，由于这只怪兽拥有极高的IQ，它觉得直接消灭小李太没有难度了，它决定要和小李在智力水平上一较高下。我们可否帮助小李来战胜强大的怪兽呢？
问题是这样的：给定一堆正整数，要求你分成两堆，两堆数的和分别为S1和S2，谁分的方案使得S1S1-S2S2的结果小（规定S1>=S2），谁就将获得胜利。
注：S2可以等于0。

输入

第一行n（1<=n<=100），表示共有n个数
第二行共n个用空格隔开的正整数ai（ai<=100），表示给定的一堆正整数。

输出

输出就一个整数，表示 S1S1-S2S2 的最小值。

样例输入

4
1 2 3 4

样例输出

0

解题思路

令S1-S2最小则结果最小，若总数和为sum，则S1、S2应最大化趋近于sum/2
则对于S2，可以转换成01背包问题求解，物品所占体积与价值均为a[i]，总容量为sum/2

AC代码

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<vector>
#include<stack>
#include<sstream>
#include<queue>
#include<map>
#include<bitset>
#include<deque>
#define ll long long
using namespace std;
int dp[105][10005];
int a[105];
int main()
{
    int n,i,j,sum=0;
    cin>>n;
    for(i=1;i<=n;i++)
    {
        cin>>a[i];
        sum+=a[i];
    }
    for(i=1;i<=n;i++)  //对于每个数
        for(j=0;j<=sum/2;j++)
        {
            if(j>=a[i])  //可以加入S2
                dp[i][j]=max(dp[i-1][j],dp[i-1][j-a[i]]+a[i]); 
            else
                dp[i][j]=dp[i-1][j];
        }
    //cout<<dp[n][sum/2]<<endl;
    cout<<sum*(sum-dp[n][sum/2]*2)<<endl;
    return 0;
}