https://leetcode-cn.com/problems/rabbits-in-forest/
思路:贪心或者脑筋急转弯?考虑当有
n
n
n只兔子说还有
m
m
m只兔子与自己有相同的颜色,当
n
<
=
m
+
1
n<=m+1
n<=m+1时,显然让这些兔子都是同一个颜色是最赚的,那么对答案的贡献就是
m
+
1
m+1
m+1(可以认为此时还有m+1-n只兔子的回答没有包含在数组中);当
n
>
m
+
1
n>m+1
n>m+1时,显然我们不能让这些兔子全部是一个颜色,只能让他们按照
m
+
1
m+1
m+1个一组进行分组。这两种情况可以结合起来:
a
n
s
+
=
⌈
n
/
(
m
+
1
)
⌉
∗
(
m
+
1
)
ans+=\lceil n/(m+1) \rceil *(m+1)
ans+=⌈n/(m+1)⌉∗(m+1)。
class Solution {
public:
int numRabbits(vector<int>& answers) {
unordered_map<int,int> count;
for(int ele:answers)
++count[ele];
int ans=0;
for(auto& ele:count)
ans+=ceil(ele.second*1.0/(ele.first+1))*(ele.first+1);
return ans;
}
};