题目描述
小Q是篮球训练队的教练,篮球队新加入了N名队员,第i名队员的篮球水平值为ai。
小Q现在要把他们按照以下的要求分为A队和B队进行训练:
1、A队的队员水平值之和严格大于B队的队员水平值之和
2、对于A队中的任意一名队员,如果把他分配到B队,A队的水平值之和就会严格小于B队的水平值之和。
3、每个队员必须要加入一个队伍
小Q现在想知道有多少种方案可以按照以上要求完成分队。
输入描述:
输入包括两行, 输入的第一行为一个正整数n(2 <= N <= 50), 表示队员的数量。 第二行包括N个正整数 ai(1 <= ai <= 6 x 104), 表示每名队员的篮球水平值, 以空格分割。
输出描述:
输出一个正整数, 表示方案数。
示例1
输入
4 5 4 7 6
输出
2
备注:比较好的一道题,这里参考了我之渺小大佬的解法。
思路:根据题意,A堆总重量严格大于B,但移动A堆任意一个苹果到B,A堆总重量严格小于B,说明,A和B的重量差一定小于A里面苹果重量的最小值的2倍,我们可以从大到小的枚举A堆里的最小值,假设枚举到w[i],比w[i]小的肯定都在B那里,比w[i]大的就开始01背包,01背包的结果出来之后,统计一下sum(A)-sum(B)小于2 * w[i]有多少种情况,如果我们按照能力值arry从大到小排序,那么枚举过程相当于一次01背包。
import java.util.*;
public class Main{
public static void main(String[] args){
int n;
int sum=0;
long ans=0;
Scanner in=new Scanner(System.in);
n=in.nextInt();
int[] a=new int[n];
for(int i=0;i<n;i++){
a[i]=in.nextInt();
sum+=a[i];
}
Arrays.sort(a);
int[] dp=new int[sum+1];
dp[0]=1;
for(int i=n-1;i>=0;i--){
for(int j=sum;j>=a[i];j--){
dp[j]+=dp[j-a[i]];
if(j>sum-j && j-a[i]<sum-j+a[i])
ans+=dp[j-a[i]];
}
}
System.out.println(ans);
}
}