姓氏 （C++）简单贪心

题目描述：

在一个很大的课室里，里面有很多学生在听课。L老师挑选了其中的N个不同的学生起立回答问题，L老师对起立的每一个学生都是问同样的问题：“在本课室里，和你同姓的学生有多少人（不包括你自己）？”。这N个起立的同学的回答如下：

第1位起立的同学回答：和我同姓的，除了我之外，本课室里还有A[1]个学生。

第2位起立的同学回答：和我同姓的，除了我之外，本课室里还有A[2]个学生。

第3位起立的同学回答：和我同姓的，除了我之外，本课室里还有A[3]个学生。

。。。。。。

第N位起立的同学回答：和我同姓的，除了我之外，本课室里还有A[N]个同学。

假设这N位同学的回答都是正确的，那么课室里至少有多少个学生？

输入格式

第一行，一个整数N。

 第二行，N个整数，空格分开，第i个整数表示A[i]。

对于20%的数据， N = 2，0<=A[i]<=10。

 对于另外20%的数据， N = 3， 0<=A[i]<=100。

 对于100%的数据，3<N<=50，0<=A[i]<=1000000。

输出格式

 一个整数，表示课室里至少有多少个学生。

输入/输出例子1

输入：

4

1  2  1  2 

输出：

5

输入/输出例子2

输入：

9

2  2  44  2  2  2  444  2  

输出：

499

样例解释

无

代码：

#include <bits/stdc++.h>
using namespace std;
long long n,a[1000010],x,s;
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
   {
	    cin>>x;
	    a[x]++;
    }
	for(int i=0;i<=1000000;i++)
	{
		if(a[i]%(i+1)!=0)
		{
			a[i]=a[i]+(i+1)-a[i]%(i+1);
		}
		s=s+a[i];
	}
	cout<<s;
	return 0;
}

哼，就知道抄代码，也不知道点赞加关注！