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) {
        MapInteger,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;
    }
最新回复(0)
/jishuDHxBZKd0mxmHoFJg1bTkkaxzdmgHlTLO3yCCCwof1I8_3D4858374
8 简首页