Leetcode 781 森林中的兔子

Leetcode 781 森林中的兔子

森林中，每个兔子都有颜色。其中一些兔子（可能是全部）告诉你还有多少其他的兔子和自己有相同的颜色。我们将这些回答放在 answers 数组里。

返回森林中兔子的最少数量。

示例:
输入: answers = [1, 1, 2]
输出: 5
解释:
两只回答了 “1” 的兔子可能有相同的颜色，设为红色。
之后回答了 “2” 的兔子不会是红色，否则他们的回答会相互矛盾。
设回答了 “2” 的兔子为蓝色。
此外，森林中还应有另外 2 只蓝色兔子的回答没有包含在数组中。
因此森林中兔子的最少数量是 5: 3 只回答的和 2 只没有回答的。

输入: answers = [10, 10, 10]
输出: 11

输入: answers = []
输出: 0

思考过程

看到题目，按照实例做了一下，又自己写了几组测试数据进行测试，如1,2,2,2,2,3,3,3,3；发现回答1的兔子，在所有回答的兔子中没有兔子可以和它匹配，所以和该兔子颜色相同的还有一只，回答2的兔子有四只，前三只回答2的可以放在一组，因为它们正好满足了颜色互相相同，剩下的一只兔子就只能自己组队了；回答为3的兔子正好四只，他们可以组成一队。到此为止我们发现回答n的兔子可以和n+1只兔子组成一队，如回答3的兔子正好4只，组成一队，无法组队的兔子要单独考虑一下，森林中还有n只和它一样颜色，所以森林中有n+1只这样颜色的兔子。

代码实现

int numRabbits(vector<int>& answers) {
	if(answers.size()==0) return 0;
	vector<int> cnt(1000,0);//该数组用来统计回答相同的兔子个数
	for(auto i:answers) cnt[i]++;
	int ans=0;
	for(int i=0;i<1000;++i){
		if(cnt[i] && cnt[i]<=(i+1)){
			ans+=(i+1);
		}
		else if(cnt[i] && cnt[i]>(i+1)){
			while(cnt[i]>(i+1)){
				ans+=(i+1);
				cnt[i]-=(i+1);
			}
			ans+=(i+1);
		}
		else{
			continue;
		}
	}
	return ans;
}