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;
}