546B. Soldier and Badges

B. Soldier and Badges
time limit per test
3 seconds
memory limit per test
256 megabytes
input
standard input
output
standard output

Colonel has n badges. He wants to give one badge to every of his n soldiers. Each badge has a coolness factor, which shows how much it's owner reached. Coolness factor can be increased by one for the cost of one coin.

For every pair of soldiers one of them should get a badge with strictly higher factor than the second one. Exact values of their factors aren't important, they just need to have distinct factors.

Colonel knows, which soldier is supposed to get which badge initially, but there is a problem. Some of badges may have the same factor of coolness. Help him and calculate how much money has to be paid for making all badges have different factors of coolness.

Input

First line of input consists of one integer n (1 ≤ n ≤ 3000).

Next line consists of n integers ai (1 ≤ ai ≤ n), which stand for coolness factor of each badge.

Output

Output single integer — minimum amount of coins the colonel has to pay.

Sample test(s)
input
4
1 3 1 4
output
1
input
5
1 2 3 2 5
output
2
Note

In first sample test we can increase factor of first badge by 1.

In second sample test we can increase factors of the second and the third badge by 1.



开两个队列,一个放重复的数字,一个放空缺的数字,然后一个个比对就行了。

#include <iostream>
#include <algorithm>
#include <queue> 
using namespace std;

int a[3000+100]={0};
queue<int>w,e;

int main(void)
{
	int n;
	cin>>n;
	for(int i=0 ; i<n ; ++i)
	{
		cin>>a[i];
	}
	sort(a,a+n);
	for(int i=1 ; i<n ; ++i)
	{
		if(a[i]==a[i-1])
			w.push(a[i]);
		else if(a[i]-a[i-1]>1)
			for(int j=a[i-1]+1 ; j<a[i] ; ++j)e.push(j);
	}
	int ans=0;
	while(!e.empty() && !w.empty())
	{
		if(e.front()>w.front())
		{
			ans+=e.front()-w.front();
			w.pop();
		}
		e.pop();
	}
	int p=a[n-1]+1;
	while(!w.empty())
	{
		ans+=p-w.front();
		w.pop();
		p++;
	}
	cout<<ans<<"\n";
	return 0;
}


你可能感兴趣的:(C++,算法,ACM)