Submission #370031

#	Time	Username	Problem	Language	Result	Execution time	Memory
370031		danielcm585	Discharging (NOI20_discharging)	C++14	11 / 100	140 ms	11372 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;
    // st.push(0);
    ll ans = 0, maxi = 0, ls = 0;
    for (int i = 1; i <= n; i++) {
        if (t[i] > maxi) {
            // int cost = t[i]*(n-i+1);
            // while (!st.empty() && (t[i]-t[st.top()])*(n-st.top()+1) < cost) {
            //     cost = (t[i]-t[st.top()])*(n-st.top()+1);
            //     st.pop();
            // }
            // ans += cost;
            // st.push(i);
            if (ls && (t[i]-t[ls])*(n-ls+1) < t[i]*(n-i+1)) ans += (t[i]-t[ls])*(n-ls+1);
            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