Submission #1233361

#TimeUsernameProblemLanguageResultExecution timeMemory
1233361TimoshDistributing Candies (IOI21_candies)C++20
3 / 100
5095 ms11336 KiB
#include "bits/stdc++.h"
#include "candies.h"

using namespace std;

struct node
{
    int lz;
};

vector<node> t;
vector<int> C;
vector<int> ans;
int n;

void push(int l, int r, int i)
{
    if (l != r && t[i].lz)
    {
        if (t[i * 2].lz)
            push(l, (l + r) / 2, i * 2);
        if (t[i * 2 + 1].lz)
            push((l + r) / 2 + 1, r, i * 2 + 1);
        t[i * 2].lz += t[i].lz;
        t[i * 2 + 1].lz += t[i].lz;
    }
    if (l == r)
        t[i].lz = max(0, min(t[i].lz, C[l]));
    else
        t[i].lz = 0;
}

void upd(int l, int r, int x, int tl = 0, int tr = n - 1, int i = 1)
{
    push(tl, tr, i);
    if (l > tr || tl > r)
        return;
    if (l <= tl && tr <= r)
    {
        t[i].lz += x;
        push(tl, tr, i);
        return;
    }
    int m = (tl + tr) / 2;
    upd(l, r, x, tl, m, i * 2);
    upd(l, r, x, m + 1, tr, i * 2 + 1);
}

void get(int tl = 0, int tr = n - 1, int i = 1)
{
    int m = (tl + tr) / 2;
    push(tl, tr, i);
    if (tl == tr)
        ans[tl] = t[i].lz;
    else
    {
        get(tl, m, i * 2);
        get(m + 1, tr, i * 2 + 1);
    }
}

vector<int> distribute_candies(vector<int> c, vector<int> l, vector<int> r, vector<int> v)
{
    n = c.size();
    C = c;
    t.resize(4 * n);
    ans.resize(n);
    for (int i = 0; i < l.size(); i++)
        upd(l[i], r[i], v[i]);
    get();
    return ans;
}
#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...