Submission #1067383

#	Time	Username	Problem	Language	Result	Execution time	Memory
1067383		golf	Distributing Candies (IOI21_candies)	C++17	11 / 100	77 ms	14644 KiB

candies

#include "candies.h"
#include <bits/stdc++.h>
using namespace std;

vector<int> distribute_candies(
    vector<int> c, vector<int> l, 
    vector<int> r, vector<int> v
) {
    int n = c.size();
    int q = v.size();

    if (q <= 2000 && n <= 2000) {
        vector<int> s(n);

        for (int i = 0; i < q; i++) {
            for (int j = l[i]; j <= r[i]; j++) {
                s[j] = max(min(s[j] + v[i], c[j]), 0);
            }
        }

        return s;
    }

    bool pos = true;
    for (int i = 0; i < n; i++) {
        if (c[i] < 0) {
            pos = false;
            break;
        }
    }

    if (pos) {
        // difference array
        vector<long long> d(n + 1);

        for (int i = 0; i < q; i++) {
            d[l[i]] += (long long)v[i];
            d[r[i] + 1] -= (long long)v[i];
        }

        vector<long long> s(n);
        s[0] = d[0];
        for (int i = 1; i < n; i++) {
            s[i] = s[i - 1] + d[i];
        }

        for (int i = 0; i < n; i++) {
            s[i] = min(s[i], (long long)c[i]);
            // s[i] = max(s[i], 0LL);
        }

        vector<int> res(n);
        for (int i = 0; i < n; i++) {
            res[i] = s[i];
        }
        return res;
    }

    vector<int> s(n, 0);
    return s;
}

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