Submission #535413

#	Time	Username	Problem	Language	Result	Execution time	Memory
535413		pokmui9909	Distributing Candies (IOI21_candies)	C++17	11 / 100	101 ms	13772 KiB

candies

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

ll N, Q;
vector<int> a;

std::vector<int> distribute_candies(std::vector<int> c, std::vector<int> l,
                                    std::vector<int> r, std::vector<int> v) {
    N = c.size(), Q = l.size();
    a.resize(N);

    if(N <= 2000 && Q <= 2000){
        for(int i = 0; i < Q; i++){
            for(int j = l[i]; j <= r[i]; j++){
                    a[j] += v[i];
                    a[j] = min(a[j], c[j]);
                    a[j] = max(a[j], 0);
            }
        }
        return a;
    } else {
        vector<ll> s(N + 1);
        for(int i = 0; i < Q; i++){
            s[l[i]]     += v[i];
            s[r[i] + 1] -= v[i];
        }
        for(int i = 1; i < N; i++){
            s[i] += s[i - 1];
        }
        for(int i = 0; i < N; i++){
            a[i] = min(s[i], (ll)c[i]);
        }
        return a;
    }
}

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