Submission #1307298

#TimeUsernameProblemLanguageResultExecution timeMemory
1307298opeleklanosDistributing Candies (IOI21_candies)C++20
8 / 100
106 ms13736 KiB
#include <iostream>
#include <vector>
#include <algorithm>
#include "candies.h"
using namespace std;

#define ll long long

ll n;
vector<ll> segTree;
vector<ll> s;

void update(ll indx, ll l, ll r, ll a, ll b, ll x){
    if(a<=l && r<=b){
        segTree[indx] += (ll)x;
        return;
    }
    ll mid = (l+r)/2;
    if(a<=mid) update(indx*2+1, l, mid, a, min(mid, b), x);
    if(b>mid) update(indx*2+2, mid+1, r, max(mid+1, a), b, x);
}

void buildS(ll indx, ll l, ll r){
    if(l == r && l<n && l>=0){
        s[l] = segTree[indx];
        return;
    }
    ll mid = (l+r)/2;
    segTree[indx*2+1] += segTree[indx];
    segTree[indx*2+2] += segTree[indx];
    segTree[indx] = -1;
    buildS(indx*2+1, l, mid);
    buildS(indx*2+2, mid+1, r);
    return;
}

vector<int> distribute_candies(vector<int> c, vector<int> l, vector<int> r, vector<int> v){
    s.assign(c.size(), 0);
    n = c.size();
    segTree.assign(n*3, 0);
    for(ll i = 0; i<l.size(); i++){
        update(0, 0, n-1, l[i], r[i], v[i]);
    }
    buildS(0, 0, n-1);
    vector<int> ans(n, 0);
    for(ll i = 0; i<n; i++) ans[i] = (int)min(s[i], (ll)c[i]);
    return ans;
}

// int main(void){
//     freopen("input.txt", "r", stdin);
//     ll ni, qu;
//     cin>>ni>>qu;
//     vector<int> c1(ni), l1(qu), r1(qu), v1(qu);
//     for(ll i = 0; i<ni; i++) cin>>c1[i];
//     for(ll i = 0; i<qu; i++) cin>>l1[i]>>r1[i]>>v1[i];
//     // for(ll i = 0; i<qu; i++) cin>>r1[i];
//     // for(ll i = 0; i<qu; i++) cin>>v1[i];
//     vector<int> ans = distribute_candies(c1, l1, r1, v1);
//     for(auto i : ans) cout<<i<<endl;
// }
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...