Submission #596762

#	Time	Username	Problem	Language	Result	Execution time	Memory
596762		serize	Distributing Candies (IOI21_candies)	C++17	3 / 100	110 ms	8052 KiB

candies

#include "candies.h"
#include <bits/stdc++.h>
#include <cassert>
#include <cstdio>
#include <vector>

using namespace std;

std::vector<int> distribute_candies(std::vector<int> c, std::vector<int> l,
                                    std::vector<int> r, std::vector<int> v) {
    int n = c.size();
    int q = l.size();
    std::vector<int> s(n);
    vector<int> abi(n+2);
    if(q <= 2000){
        for(int i = 0; i < q; i++){
            for(int j = l[i]; j <= r[i]; j++){
                if(v[i] < 0) s[j] = max(0,s[j]+v[i]);
                else s[j] = min(c[j],s[j]+v[i]);
            }
        }
    }
    else{
        for(int i = 0; i < q; i++){
            int k = l[i]+1;
            while(k < (n+2) ){
                abi[k] += v[i];
                k += k&(-k);
            }
            k = r[i]+2;
            while(k < (n+2)){
                abi[k] += (-v[i]);
                k += k&(-k);
            }
        }
        for(int i = 0; i < n; i++){
            int x = 0, k = i+1;
            while(k > 0){
                x += abi[k];
                k -= k&(-k);
            }
            s[i] = min(c[i],x);
        }
    }
    return s;
}

• Subtask 1: 3 / 3
• Subtask 2: 0 / 8
• Subtask 3: 0 / 27
• Subtask 4: 0 / 29
• Subtask 5: 0 / 33