力扣练习——67 森林中的兔子

67 森林中的兔子

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

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

示例:

输入: answers = [1, 1, 2]

输出: 5

解释:

两只回答了 “1” 的兔子可能有相同的颜色，设为红色。

之后回答了 “2” 的兔子不会是红色，否则他们的回答会相互矛盾。

设回答了 “2” 的兔子为蓝色。

此外，森林中还应有另外 2 只蓝色兔子的回答没有包含在数组中。

因此森林中兔子的最少数量是 5: 3 只回答的和 2 只没有回答的。

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

输出: 11

输入: answers = []

输出: 0

2.输入说明
首先输入answers数组的长度n，n<=1000

然后输入n个整数，以空格分隔，每个整数在 [0, 999] 范围内
3.输出说明
输出一个整数
4.范例
输入
3
1 1 2
输出
5
5.代码

#include<iostream>
#include<map>
#include<string>
#include<unordered_map>
#include<algorithm>
#include<string.h>
#include<sstream>
#include <vector>

using namespace std;

int NumOfRabbits(vector<int> answers)
{

	
	//    x/y  向上取整的写法  :(x+(y-1))/y    或者直接用ceil(x/y);
	//    x/(y+1)    向上取整  :(x+y)/(y+1)
	//参考贪心算法题解: https://leetcode.cn/problems/rabbits-in-forest/solution/sen-lin-zhong-de-tu-zi-by-leetcode-solut-kvla/
	if (answers.size() == 0)
		return 0;
	int n = answers.size();
	unordered_map<int, int>map;
	for (auto t : answers)
	{
		map[t]++;//统计每个t出现的次数
	}
	int ans = 0;
	for (auto s : map)
	{
		//同一颜色的兔子回答的数值必然是一样的
		//但回答同样数值的，不一定就是同颜色兔子

		int val = s.first;//回答相同颜色的兔子只数  ，若回答val，则至少有（val+1）只兔子是同种颜色的
		int cnt = s.second;
		if (cnt % (val + 1) == 0)
			ans += (cnt / (val + 1))*(val + 1);//cnt/(val+1)计算出最少有几种颜色    每种颜色至少有val+1只兔子
		else
			ans+= (cnt / (val + 1)+1)*(val + 1);
	}
	return ans;
}



int main()

{
	int n,tmp;
	cin >> n;
	vector<int>answers;
	for (int i = 0; i < n; i++)
	{
		cin >> tmp;
		answers.push_back(tmp);
	}
	int res = NumOfRabbits(answers);
	cout << res << endl;
	return 0;

}