781. 森林中的兔子

难度 medium
森林中，每个兔子都有颜色。其中一些兔子（可能是全部）告诉你还有多少其他的兔子和自己有相同的颜色。我们将这些回答放在 answers 数组里。

返回森林中兔子的最少数量。

示例:
输入: answers = [1, 1, 2]
输出: 5
解释:
两只回答了 "1" 的兔子可能有相同的颜色，设为红色。
之后回答了 "2" 的兔子不会是红色，否则他们的回答会相互矛盾。
设回答了 "2" 的兔子为蓝色。
此外，森林中还应有另外 2 只蓝色兔子的回答没有包含在数组中。
因此森林中兔子的最少数量是 5: 3 只回答的和 2 只没有回答的。

输入: answers = [10, 10, 10]
输出: 11

输入: answers = []
输出: 0
说明:

answers 的长度最大为1000。
answers[i] 是在 [0, 999] 范围内的整数。

解题思路：这道题还是相对来说比较简单的，只要想明白了，假设回答2的兔子有1，2，3只，从最少数量的要求来考虑，这几只兔子最好属于同一个集合，而一旦超过3，比如有4或5只，那肯定就不能归在同一个集合里了，只能属于另一个集合，因此，如果有n只兔子回答了m，那至少可以划分为ceil(float(n/(m+1)))个集合，每个集合有(m+1)只兔子，因此我们可以用一个hashmap来保存我们遍历到的answers中兔子的回答和这个回答的兔子数目，然后用上面的方法计算每种回答的兔子的最少数目，并且累加，就可以得到结果。

代码 t50 s39 java

class Solution {
    public int numRabbits(int[] answers) {
        HashMap<Integer, Integer> mp = new HashMap<>();
        int res = 0, val = 0;
        for(int num : answers){
            if(!mp.containsKey(num)){
                mp.put(num, 1);
            }else{
                val = mp.get(num);
                mp.put(num, val + 1);
            }
        }
        for (Map.Entry<Integer, Integer> entry : mp.entrySet()) {
            // System.out.println("key: " + entry.getKey() + " val: " + entry.getValue());
            res += Math.ceil((float)entry.getValue() / (entry.getKey() + 1)) * (entry.getKey() + 1);
		}        
        return res;
    }
}

参考资料

标签：781,回答,answers,num,mp,兔子,entry,森林
来源： https://www.cnblogs.com/zhengxch/p/14615623.html