Submission #960001

#	Time	Username	Problem	Language	Result	Execution time	Memory
960001		Cookie	Feast (NOI19_feast)	C++14	100 / 100	263 ms	27588 KiB

feast

#include<bits/stdc++.h>
#include<fstream>
using namespace std;
#define sz(a) (int)a.size()
#define ALL(v) v.begin(), v.end()
#define ALLR(v) v.rbegin(), v.rend()
#define ll long long
#define pb push_back
#define forr(i, a, b) for(int i = a; i < b; i++)
#define dorr(i, a, b) for(int i = a; i >= b; i--)
#define ld long double
#define vt vector
#include<fstream>
#define fi first
#define se second
#define pll pair<ll, ll>
#define pii pair<int, int>
#define mpp make_pair
const ld PI = 3.14159265359;
using u128 = __uint128_t;
const int x[4] = {2, -2, 2, -2};
const int y[4] = {2, 2, -2, -2};
const ll mod = 1e9 + 7;
const int mxn = 1e6 + 69, mxq = 1e5 + 5, sq = 500, mxv = 2e7 + 5;
//const int base = (1 <<18);
const ll inf = 1e18 + 5, neg = -69420;
mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count());
//const int x[9] = {0, 1, 1, -1, -1, 2, -2, 2, -2};
//const inty[9] = {0, 2, -2, 2, -2, 1, 1, -1, -1};
int n, k;
ll a[mxn + 1], val[mxn + 1];
signed main(){
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    //freopen("COLLEGE.INP", "r", stdin);
    //freopen("COLLEGE.OUT", "w", stdout);
    cin >> n >> k;
    vt<ll>comp;
    a[0] = -inf; a[n + 1] = -inf;
    for(int i = 1; i <= n; i++){
        cin >> a[i];
    }
    int cnt = 0;
    for(int i = 0; i <= n + 1;){
        ll sm = 0;
        int r = i;
        while(r <= n + 1 && (a[r] >= 0) == (a[i] >= 0)){
            sm += a[r++];
        }
        comp.pb(sm);
        if(sm >= 0){
            cnt++;
        }
        i = r;
    }
    for(int i = 0; i < sz(comp); i++)val[i] = comp[i];
    set<pair<ll, ll>>lar; set<int>pos;
    for(int i = 0; i < sz(comp); i++){
        lar.insert(mpp(abs(comp[i]), i)); pos.insert(i);
    }
    for(int i = cnt; i > k; i--){

        auto [va, id] = *lar.begin();
        auto it = pos.lower_bound(id), lw = prev(it), hg = next(it);
        lar.erase(mpp(abs(val[*hg]), *hg)); lar.erase(mpp(abs(val[*lw]), *lw));lar.erase(mpp(abs(val[id]), id));
        val[id] += val[*lw]; val[id] += val[*hg];
        pos.erase(*lw); pos.erase(*hg);  lar.insert(mpp(abs(val[id]), id));
    }
    ll ans = 0;
    for(auto it: lar){
        if(val[it.se] >= 0)ans += val[it.se];
    }
    cout << ans;
    return(0);
}

Compilation message (stderr)

feast.cpp: In function 'int main()':
feast.cpp:62:14: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   62 |         auto [va, id] = *lar.begin();
      |              ^

• Subtask 1: 4 / 4
• Subtask 2: 8 / 8
• Subtask 3: 18 / 18
• Subtask 4: 10 / 10
• Subtask 5: 11 / 11
• Subtask 6: 20 / 20
• Subtask 7: 29 / 29