[2021.11.21]UPC-2021级新生个人训练赛第6场-5785 Problem F 小李打怪兽I

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

输入

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

输出

输出就一个整数，表示 S1*S1-S2*S2 的最小值。

样例输入 Copy

4 
1 2 3 4

样例输出 Copy

0

看了下面这个链接后应该就知道怎么做了~  ; )

动态规划之背包问题系列 - 知乎 (zhihu.com)

分享一下我的代码 : ) ：

#include <algorithm>
#include <cstdio>
using namespace std;
int main(){
    long long int n,i,s=0,ans=0,j;
    long long int a[110];
    scanf("%lld",&n);
    for(i=1;i<=n;i++){
        scanf("%lld",&a[i]);
        s+=a[i]; //s表示所有数的总和
    }
    long long int dp[10005]={0};
    for(i=1;i<=n;i++){
        for(j=s/2;j>=a[i];j--){     //要把数分成两堆，实际上相当于最大负重是s/2的背包问题~
            dp[j]=max(dp[j],dp[j-a[i]]+a[i]);
        }
    }
    for(j=s/2;j>=0;j--)
        ans=max(dp[j],ans);    //ans<=s/2所以肯定是两堆数中较小的那一堆，除非两堆数各自的和相等~
    printf("%lld",(s-ans)*(s-ans)-ans*ans);
    return 0;
}