$O(n)$

每个人取得的糖果数量受两个条件限制，设第 $i$ 个人取得的糖果数为 $ans[i]$.

1. 至少要 $a[i]$个：$ans[i] >= a[i]$
2. 比前面的人多: $ans[i] \geq ans[i - 1] + 1$

第 $i$ 个人取得 $max(ans[i - 1] + 1, a[i])$ 为最小值

#include<bits/stdc++.h>

using namespace std;

typedef long long LL;
typedef pair<LL, LL> PII;

#define x first
#define y second

const int N = 5e3 + 10, mod = 1e9 + 7;

int a[N];

int main()
{
    int n;
    cin >> n;
    LL ans = 0;
    for(int i = 1; i <= n; i ++)
    {
        cin >> a[i];
        a[i] = max(a[i - 1] + 1, a[i]);
        ans += a[i];
    }
    cout << ans << endl;
}