网易历届笔试题（11）

最新推荐文章于 2022-04-02 12:02:45 发布

信仰..

最新推荐文章于 2022-04-02 12:02:45 发布

阅读量514

点赞数

分类专栏：常见笔试编程题

本文链接：https://blog.csdn.net/haut_ykc/article/details/116243425

版权

常见笔试编程题专栏收录该内容

32 篇文章 11 订阅

订阅专栏

本文介绍了一道关于如何将篮球队员按能力值分成A、B两队的题目，通过01背包算法巧妙地解决了满足特定条件的分队方案计数问题。关键在于理解A、B队实力差的限制条件，并利用排序和动态规划求解多样化的分队组合。

摘要由CSDN通过智能技术生成

题目描述

小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

输出

备注：比较好的一道题，这里参考了我之渺小大佬的解法。

思路：根据题意，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);
    }
}