C ++中的森林兔子
假设在森林中,每只兔子都有一些颜色。现在,兔子的一些子集(可能是全部)将告诉我们还有多少只兔子具有相同的颜色。这些答案放置在一个数组中。我们必须找到可能在森林中的兔子的最小数量。因此,如果输入类似于[1,1,2],则输出将为5,因为两只回答“1”的兔子可能都是相同的颜色,例如白色。现在,兔子所回答的“2”不能为白色,否则答案将不一致。说回答“2”的兔子是黑色的。然后,在森林中应该有另外2只未对阵列做出反应的黑兔子。因此,森林中尽可能少的兔子数量是5:3回答了,再加上2没有回答。
为了解决这个问题,我们将遵循以下步骤-
制作一个映射m,n:=数组ans的大小
设置ret为0
对于i,范围为0至n–1
将m[x]增加1
如果m[x]=x,则从m中删除x
x:=ans[i]
如果x=0,则将ret增加1,并跳过此迭代的下一部分。
如果m具有x,则将ret增加(x+1),设置m[x]:=0
除此以外
返回ret
让我们看下面的实现以更好地理解-
示例
#include <bits/stdc++.h>
using namespace std;
class Solution {
public:
int numRabbits(vector<int>& ans) {
map <int, int> m;
int n = ans.size();
int ret = 0;
for(int i = 0; i < n; i++){
int x = ans[i];
if(x == 0){
ret++;
continue;
}
if(!m.count(x)){
ret += (x + 1);
m[x] = 0;
}else{
m[x]++;
if(m[x] == x){
m.erase(x);
}
}
}
return ret;
}
};
main(){
vector<int> v = {1,1,2};
Solution ob;
cout << (ob.numRabbits(v));
}输入值
[1,1,2]
输出结果
5