Submission #834772

#	Time	Username	Problem	Language	Result	Execution time	Memory
834772		Baytoro	Distributing Candies (IOI21_candies)	C++17	3 / 100	5037 ms	13004 KiB

candies

#include "candies.h"
//#include "grader.cpp"
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define all(x) x.begin(), x.end()

long long tree[1000005];
void add(int r, int v){
	r++;
	for(;r<400005;r+=r&-r)
		tree[r]+=v;
}
void add(int l, int r, int v){
	add(r,v);
	if(l-1>=0)
		add(l-1,v);
}
int get(int x){
	x++;
	int sum=0;
	for(;x>=0;x-=x&-x)
		sum+=tree[x];
	return sum;
}
int get(int r, int val){
	return get(r)-get(r-1);
}
vector<int> distribute_candies(vector<int> c, vector<int> l,
                                    vector<int> r, vector<int> v) {
    int n=c.size(),q=v.size();
    vector<int> a(n);
    if(n<=2000 && q<=2000){
		for(int t=0;t<q;t++){
			for(int i=l[t];i<=r[t];i++){
				a[i]+=v[t];
				a[i]=min(c[i],a[i]);
				a[i]=max(a[i],0);
			}
		}
	}
	else{
		for(int t=0;t<q;t++){
			add(l[t],r[t],v[t]);
		}
		for(int i=0;i<n;i++)
			a[i]=max(c[i],get(i,0));
	}
	return a;
}

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