#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<queue>
#include<vector>
#include<map>
#include<unordered_map>
#include<string>
#include<cctype>
#include<stack>
#include<unordered_set>
using namespace std;
/*
题意:森林中,每个兔子都有颜色。其中一些兔子(可能是全部)告诉你还有多少其他的兔子和自己有相同的颜色。我们将这些回答放在 answers 数组里。
返回森林中兔子的最少数量。
分析:报数相同的兔子肯定是可以是相同的颜色,报数不同的兔子一定是不同的颜色,
可以用set记录所有的肯定不同颜色的兔子的数量
注意:
1.如果兔子报数为0,说明自己与众不同
2.如果报的数(假设为a)相同的数量超过了a+1,则一定代表不同颜色,超过2*(a+1)同理可推
*/
int numRabbits(vector<int>& answers) {
unordered_map<int,int> se;
int ans = 0;
for (int i = 0; i < answers.size(); ++i) {
se[answers[i] + 1]++;
}
for (auto it : se) {
int k = it.second / it.first;
if (it.second%it.first) k++;
ans += k * it.first;
}
return ans;
}
int main() {
vector<int> b = { 0,0,1,1,1 };
cout << numRabbits(b);
return 0;
}
781
最新推荐文章于 2024-01-14 16:51:43 发布