1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int main() {
    // Optymalizacja operacji wejścia/wyjścia
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);

    int n;
    long long k;
    if (!(cin >> n >> k)) return 0;

    vector<long long> a(n);
    for (int i = 0; i < n; ++i) {
        cin >> a[i];
    }

    // Tablica przechowująca najmniejsze dopuszczalne wysokości 
    // z uwagi na ograniczenia od lewej strony
    vector<long long> P(n);
    P[0] = a[0];
    for (int i = 1; i < n; ++i) {
        P[i] = max(a[i], P[i - 1] - k);
    }

    // Tablica przechowująca najmniejsze dopuszczalne wysokości 
    // z uwagi na ograniczenia od prawej strony
    vector<long long> S(n);
    S[n - 1] = a[n - 1];
    for (int i = n - 2; i >= 0; --i) {
        S[i] = max(a[i], S[i + 1] - k);
    }

    long long total_trucks = 0;
    for (int i = 0; i < n; ++i) {
        // Docelowa, optymalna wysokość elementu i-tego
        long long target_height = max(P[i], S[i]);
        total_trucks += (target_height - a[i]);
    }

    // Wypisanie wyniku
    cout << total_trucks << "\n";

    return 0;
}