网易面试题 糖果谜题

[编程题] 糖果谜题

时间限制：1秒

空间限制：32768K

小明是幼儿园的一名老师。某天幼儿园园长给小朋友们每人发一颗糖果，小朋友们拿到后发现有一些同学拿到的糖果颜色和自己相同，有一些同学糖果颜色和自己不同。

假定每个小朋友只知道有多少同学和自己拿到了相同颜色的糖果。

上课后，有一部分小朋友兴奋的把这一结果告诉小明老师，并让小明老师猜一猜，最少有多少同学拿到了糖果。

例如有三个小朋友告诉小明老师这一结果如下：

其中第一个小朋友发现有1人和自己糖果颜色一样，第二个小朋友也发现有1人和自己糖果颜色一样，第三个小朋友发现有3人和自己糖果颜色一样。

第一二个小朋友可互相认为对方和自己颜色相同，比如红色；

第三个小朋友不可能再为红色（否则第一二个小朋友会发现有2人和自己糖果颜色相同），假设他拿到的为蓝色糖果，那么至少还有另外3位同学拿到蓝色的糖果，最终至少有6位小朋友拿到了糖果。

现在请你帮助小明老师解答下这个谜题。

 

输入描述:

假定部分小朋友的回答用空格间隔，如 1 1 3

 

输出描述:

直接打印最少有多少位小朋友拿到糖果
如 6

 

输入例子1:

1 1 3

 

输出例子1:

6

 

输入例子2:

0 0 0

 

输出例子2:

3

 

例子说明2:

三位小朋友都没发现有人和自己的颜色相同，所以最少的情况就是三位小朋友糖果的颜色均不同

/*
分析:通过map来记录相同颜色的数量，并通过数量num与val+1进行比较来计算。注意当num不足到一个
val+1时，只算上一个val+1.
注意:只是部分同学告诉老师，所以例如2 2的时候也是算最少三个糖果
*/
#include<iostream>
#include<map>
#include<stdio.h> 
#include<vector>  
using namespace std;

int main()
{

    map<int,int> myMap;  
    int res = 0;
    int a;
    while(cin>>a){
    	myMap[a]+=1;
    	char ch = getchar();//通过getchar()来判断最后输入回车符结束 
    	if(ch == '\n')
    		break;
    }
   
   	for(auto it:myMap){
   		int val = it.first;
   		int num = it.second;
   		
   		if(num <= val)
   			res += val+1;
   		else{
   			res += num/(val+1)*(val+1);
   			if( num%(val+1) != 0)//不足的只算一次 
   			res += (val+1);
   		}
   	}
    
    cout<<res<<endl;
    return 0;
}