Submission #618097

#	Time	Username	Problem	Language	Result	Execution time	Memory
618097		BalintR	Distributing Candies (IOI21_candies)	C++17	38 / 100	5076 ms	7372 KiB

candies

#include <vector>
using namespace std;
#pragma GCC target "avx2"
#pragma GCC optimize "Ofast"

typedef vector<int> vi;
#define SZ(v) ((int) (v).size())
#define min(a, b) ((a)<(b)?(a):(b))
#define max(a, b) ((a)<(b)?(b):(a))

const int MN = 2e5;
const int BSZ = 2e3;
int n, q;
int arr[BSZ], cap[BSZ];

vi distribute_candies(vi cIn, vi lIn, vi rIn, vi vIn){
    n = SZ(cIn);
    q = SZ(lIn);
    vi res(n);

    for(int bl = 0; bl < n; bl += BSZ){
        int br = min(n, bl+BSZ);
        for(int i = bl; i < br; i++) arr[i-bl] = 0, cap[i-bl] = cIn[i];

        for(int qi = 0; qi < q; qi++){
            int l = max(0, lIn[qi]-bl), r = min(br, rIn[qi]+1)-bl, v = vIn[qi];
            if(v < 0) for(int j = l; j < r; j++) arr[j] = max(arr[j] + v, 0);
            else for(int j = l; j < r; j++) arr[j] = min(arr[j] + v, cap[j]);
        }

        for(int i = bl; i < br; i++) res[i] = arr[i-bl];
    }

    return res;
}

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