Submission #370028

#	Time	Username	Problem	Language	Result	Execution time	Memory
370028		danielcm585	Discharging (NOI20_discharging)	C++14	11 / 100	126 ms	18028 KiB

Discharging

#include <bits/stdc++.h>
using namespace std;

#define fi first
#define se second

typedef long long ll;
typedef pair<int,int> ii;

const int N = 1e6;
int n;
ll t[N+2], h[N+2];

int main() {
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    cin >> n;
    for (int i = 1; i <= n; i++) {
        cin >> t[i];
    }
    stack<ll> st;
    ll ans = 0, maxi = 0, ls = 0;
    for (int i = 1; i <= n; i++) {
        if (t[i] > maxi) {
            // cout << i << " -> " << (t[i]-t[ls])*(n-ls+1) << ' ' << t[i]*(n-i+1) << '\n';
            if (ls && (t[i]-t[ls])*(n-ls+1) < t[i]*(n-i+1)) {
                ans += (t[i]-t[ls])*(n-ls+1);
                t[ls] = t[i];
            }
            else {
                ans += t[i]*(n-i+1);
                ls = i;
            }
            maxi = t[i];
        }
    }
    cout << ans << '\n';
    return 0;
}

• Subtask 1: 0 / 9
• Subtask 2: 0 / 13
• Subtask 3: 0 / 25
• Subtask 4: 11 / 11
• Subtask 5: 0 / 14
• Subtask 6: 0 / 28