Codeforces #304 (Div. 2) B. Soldier and Badges （ 贪心

B. Soldier and Badges

题目描述

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.

输入

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 single integer — minimum amount of coins the colonel has to pay.

样例

inputCopy
4
1 3 1 4
output
1
inputCopy
5
1 2 3 2 5
output
2

题意

贪心写法 两种写法$O(n)||O(n^2)$

AC代码

$O(n)$

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

#define LL long long
#define CLR(a,b) memset(a,(b),sizeof(a))
#define ls st<<1
#define rs st<<1|1

const int MAXN = 1e5*4+10;
int arr[MAXN];
int main() {
    int n, x;
    cin >> n;
    for(int i = 1; i <= n; i++ ) {
        cin >> x;
        arr[x]++;
    }
    int res = 0;
    for(int i = 1; i <= 2*n; i++) {
        if(arr[i] > 1) {
            res += arr[i]-1;
            arr[i+1] +=  arr[i]-1;
        }
    }
    cout << res << endl;

return 0;
}

$O(n^2)$

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

#define LL long long
#define CLR(a,b) memset(a,(b),sizeof(a))
#define ls st<<1
#define rs st<<1|1

const int MAXN = 1e5*4+10;
int arr[MAXN];
int main() {
    int n, x;
    cin >> n;
    for(int i = 1; i <= n; i++ ) {
        cin >> arr[i];
    }
    sort(arr+1, arr+1+n);
    int res = 0;
    for(int i = 1; i < n; i++) {
        if(arr[i]==arr[i+1]) {
            arr[i+1] ++;
            res ++;
        }
        if(arr[i]>=arr[i+1]) {
            res = res + arr[i]-arr[i+1]+1;
            arr[i+1] = arr[i]+1;
        }
    }
    cout << res << endl;

return 0;
}