CF 546B Soldier and Badges

题意: 一个数组,其中可能有相同的数,使数组中某一个数加1的成本为1,要使数组中所有的数均不相同,所需要的成本是多少.
很简单的题,桶排序或者类似于前缀和的算法都可以.但有一个坑一开始没有注意到就是最大的数可能超过n
例如:n = 5
3 3 4 4 5
这样输入的话你要让数组变成 3 4 5 6 7 而不是 1 2 3 4 5 所以代码会出错

#include<bits/stdc++.h>

using namespace std;

long long a[8000];
int main(){
int n;
cin>>n;
long long sum= 0;
memset(a,0,sizeof(a));
for(int i = 1;i <= n;i++)
  {
  	int x;
  	scanf("%d",&x);
  	a[x]++;
  }
for(int i = 1;i < 2*n;i++)//注意这里,本来i< n,导致了不停的WA
 {
 	if(a[i]>1)
 	{
	 sum+=a[i]-1;
     a[i+1] += a[i]-1;
 }
}
 cout <<sum<<endl;
	return 0;
	
}