如图:
题目本身难度并不大,重要的是读懂题目:
根据题目描述,就是为了找出相同值的数量,然后对于任意一个值n,它能容纳的颜色相同的最大数量是n+1;多于这个数量都算另一个颜色。
所以我们考虑建立一个哈希表,记录同一颜色(相同值)的数量,对于值n,如果answers中n的数量位于[1,n+1]之间,那么这种颜色数量为n+1,如果是[n+2,2n+2]之间就是2n+2…如此我们便找到了规律,位于某个区间的值统统等于右边界值,对于值3,设数组中3的数量有m,那么标号为3的最少有多少只兔子呢?答案为:(3+m)/44;对于值n,答案为:(m+n)/(n+1)(n+1)
注:(m+n)/(n+1)是为了取整,就是看m+n中包含了多少个n+1,注意除号/,计算机结果默认是去尾了,所以对于1-n的数量/(n+1)再 *(n+1)之后就变成了0了,不等于n+1,所以+n,再/(n+1) *(n+1),结果=n+1,即使是边界值n+1+n再/(n+1) *(n+1),答案依旧等于n+1.
看代码:
class Solution {
public:
int numRabbits(vector<int>& answers) {
int sum=0;
unordered_map<int,int> num;
for(int i=0;i<answers.size();i++){
num[answers[i]]++;
}
for(auto a:num){
int m=a.second;
sum += ((m -1)/ (a.first + 1)+1) * (a.first + 1);
}
return sum;
}
};