Submission #266754

#	Time	Username	Problem	Language	Result	Execution time	Memory
266754		eohomegrownapps	Discharging (NOI20_discharging)	C++14	11 / 100	133 ms	760 KiB

Discharging

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

int main(){
    cin.tie(0);
    ios_base::sync_with_stdio(0);
    ll n;
    cin>>n;
    ll prev = 0;
    ll tot = 0;
    stack<pair<ll,ll>> s; //index, time
    ll val;
    for (ll i = 0; i<n; i++){
        cin>>val;
        if (val<=prev){
            continue;
        } else {
            prev=val;
        }
        ll ind = i;
        tot+=(n-ind)*val;
        while (s.size()>0){
            auto f = s.top();
            ll tind = f.first;
            ll th = f.second;
            //cout<<tind<<' '<<th<<'\n';
            //can we do better by popping?
            if ((n-ind)*val+(n-tind)*th>(n-tind)*val){
                //cout<<"subtract\n";
                tot-=(n-ind)*val+(n-tind)*th;
                //cout<<tot<<'\n';
                tot+=(n-tind)*val;
                //cout<<tot<<'\n';
                ind=tind;
                s.pop();
            } else {
                break;
            }
        }
        //cout<<i<<": "<<tot<<'\n';
        s.push({ind,val});
    }
    cout<<tot<<'\n';
}

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