题解
这题很有意思。
想想,有兔子说有 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;
}