c_cpp 通过最小增量使元素在排序数组中不同

Posted 2021-05-14

//https://www.geeksforgeeks.org/making-elements-distinct-sorted-array-minimum-increments/
#include <iostream>
using namespace std;

int func (int a[], int n) {
    int prev= a[0], sum= a[0];
    for (int i=1;i<n; i++) {
        if (a[i]<=prev) {
            prev+= 1;
            sum+= prev;
        }
        else {
            prev= a[i];
            sum+= a[i];
        }
    }
    return sum;
}

int main() {
    int n;
    cin>>n;
    int a[n];
    for (int i=0;i<n;i++)
        cin>>a[i];
    cout<<func(a,n);
}