2019上海网络赛 J Stone game

题意：分2堆石头，使得第一堆的总重量大于第二堆，并且去掉第一堆的任意一块，必有第一堆的重量小于等于第二堆，求分配方法的总数;

思路考虑将数组从大到小排序，然后01背包，dp[i]表示价值为i的分配方法的总数;

代码：

#include<bits/stdc++.h>
#include<iostream>
#include<cstdio>
using namespace std;
const int mod=1e9+7;
bool cmp(int a, int b)
{
return a>b;
}
int dp[150005];
int a[305];
int main()
{
int T,n;
scanf("%d", &T);
while(T--)
{
scanf("%d", &n);
int sum=0, ans=0;
for(int i=0; i<n; i++)
{
scanf("%d", &a[i]);
sum+=a[i];
}
sort(a, a+n, cmp);
memset(dp,0,sizeof(dp));
dp[0]=1;
for(int i=0; i<n; i++)
{
for(int j=sum; j>=a[i]; j--)
{
if(j>=sum-j && j-a[i]<=sum-j)
{
ans=(ans+dp[j-a[i]])%mod;
}
dp[j]=(dp[j]+dp[j-a[i]])%mod;
}
}
printf("%d\n",ans);
}
return 0;
}