LeetCode 781. Rabbits in Forest

题解

这题很有意思。
想想，有兔子说有 x 个同色兔子，那么其实就是说有 x+1 个兔子同色。
换言之，这个 x 最多可以在数组中重复 x+1 次，但是显然 x 可以出现任意次。
少于 x+1 就意味着有部分兔子没说话，多余则表示还有另外颜色的兔子
也有 x+1 个。
所以，此题用map记录 x 的次数。然后算 (x+1) * ceil( map[x]/(x+1) ）即可。

---

Code

public int numRabbits(int[] answers) {
        Map<Integer,Integer> map =new HashMap<>();
        int res=0;
        for(int a:answers){
            map.put( a , map.getOrDefault(a,0)+1);
        }
        
        for(Integer n : map.keySet()){
            int counts = map.get(n)/(n+1);
            res+=  map.get(n)%(n+1) == 0? counts*(n+1) : (counts+1)*(n+1);
        }
        
        return res;
    }