957D - Riverside Curio - CodeForces Solution

#include <iostream>
using namespace std;

int main()
{
    int arr[1000000], arr_max[1000000];
    int n;
	cin >> n;
	for (int i = 1; i <= n; i++) {
		cin >> arr[i];
	}
	arr_max[1] = 1;
	for (int i = 2; i <= n; i++) 
    {
		arr_max[i] = max(arr_max[i - 1], arr[i] + 1);
	}
    int k = n;
    while(k>=2){
        if (1 < arr_max[k] - arr_max[k - 1]) 
        {
			arr_max[k - 1] = arr_max[k] - 1;
		}
        k--;
    }
    long long result = 0;
	for (int i = 1; i <= n; i++) 
    {
		if (0 < arr_max[i] - arr[i] - 1) 
        {
			result = result + arr_max[i] - arr[i] - 1;
		}
	}
    cout << result << '\n';
    return 0;
}