【华为OD题库-020】阿里巴巴找黄金宝箱(II)-Java

题目

一贫如洗的樵夫阿里巴巴在去砍柴的路上，无意中发现了强盗集团的藏宝地，藏宝地有编号从0~N的箱子，每个箱子上面贴有箱子中藏有金币的数量。从金币数量中选出一个数字集合，并销毁贴有这些数字的每个箱子，如果能销毁一半及以上的箱子，则返回这个数字集合的最小大小。
输入描述:
一个数字字串，数字之间使用逗号分隔，例如: 6,6,6,6,3,3,3,1,1,5。字串中数字的个数为偶数，并且个数>=1, <=100000; 每个数字>=1, <=100000;
输出描述
这个数字集合的最小大小，例如: 2
示例1
输入:
1,1,1,1,3,3,3,6,6,8
输出:
2
说明:
选择集合{1,8}，销毁后的结果数组为[3,3,3,6,6]，长度为5,长度为原数组的一半。
大小为2的可行集合还有{1,3},{1,6},{3,6}。
选择{6,8}集合是不可行的，它销毁后的结果数组为[1,1,1,1,3,3,3],新数组长度大于原数组的二分之一。
示例2
输入:
2,2,2,2
输出:
1
说明:
我们只能选择集合{2}，销毁后的结果数组为空

思路

用res表示结果。boxs代表宝箱数组。根据题意，就是要销毁res种金币的宝箱，使得销毁的宝箱数量大于等于原来boxs总量的一半。
以示例1为例，宝箱boxs为：1,1,1,1,3,3,3,6,6,8；宝箱个数为10.
使用map记录每种宝箱出现的次数：1:4；3:3；6:2，8:1。
res的取值范围为[1~4]，最少可以消除1种，最多消除4种（即map.size()）
要求res的最小值，可以考虑使用二分法解决此题，伪代码如下：

while(l<r)
	if checked(mid):
		r=mid
	else
		r=mid+1
return l

其中l=1,right=map.size()。现在关键在于实现checked方法。
checked方法需要校验对于给定mid（mid代表宝箱种类数），找出来的宝箱数量是否大于等于原来宝箱数量的一半（题目说的宝箱数量为偶数，所以直接len/2即可，如果考虑奇数情况，可以用（len+1）/2，不影响）。
比如mid=2；那么需要找出2种宝箱，使他们的数量尽可能大。所以可以将map按照value从大到小排序。取头2个宝箱的数量和即可。这里只关心宝箱的数量，不关心具体哪个宝箱。所以可以把map.values()转为list后按照降序排序。

题解

package hwod;

import java.util.*;

public class FindGoldBox2 {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        String[] inputs = sc.nextLine().split(",");
        int[] boxs = Arrays.stream(inputs).mapToInt(Integer::parseInt).toArray();


        System.out.println(findGoldBox(boxs));
    }

    private static Map<Integer, Integer> map = new HashMap<>();

    private static int findGoldBox(int[] boxs) {
        int nums = boxs.length;
        for (int i = 0; i < nums; i++) {
            map.put(boxs[i], map.getOrDefault(boxs[i], 0) + 1);
        }
        int left = 1, right = map.size();
        while (left < right) {
            int mid = left + right >> 1;
            if (checked(mid, nums)) {
                right = mid;
            } else {
                left = mid + 1;
            }
        }
        return left;
    }

    private static boolean checked(int mid, int nums) {
        List<Integer> list = new ArrayList<>(map.values());
        list.sort((o1, o2) -> o2 - o1);
        int cnt = 0;
        for (int i = 0; i < mid; i++) {
            cnt += list.get(i);
        }
        return cnt>=(nums+1)/2;
    }
}

推荐

如果你对本系列的其他题目感兴趣，可以参考华为OD机试真题及题解（JAVA），查看当前专栏更新的所有题目。

说明

本专栏所有文章均为原创，欢迎转载，请注明文章出处：https://blog.csdn.net/qq_31076523/article/details/134176793。百度和各类采集站皆不可信，搜索请谨慎鉴别。技术类文章一般都有时效性，本人习惯不定期对自己的博文进行修正和更新，因此请访问出处以查看本文的最新版本。