Submission #834719

#	Time	Username	Problem	Language	Result	Execution time	Memory
834719		JakobZorz	Distributing Candies (IOI21_candies)	C++17	3 / 100	5048 ms	25764 KiB

candies

#include"candies.h"
#include<iostream>
#include<vector>
using namespace std;
typedef long long ll;

int n,days;

vector<int>vec;
void tree_init(){
    vec.resize(days);
}

void tree_add(int pos,int val){
    //for(int i=pos;i<days;i++)
        vec[pos]+=val;
}

int tree_get(int bound){
    int curr=0;
    for(int i=0;i<days;i++){
        curr+=vec[i];
        curr=max(0,min(bound,curr));
    }
    return curr;
}

vector<int>distribute_candies(vector<int>c,vector<int>l,vector<int>r,vector<int>v){
    n=(int)c.size();
    days=(int)v.size();
    vector<vector<pair<int,int>>>events;
    events.resize(n+1);
    
    for(int i=0;i<days;i++){
        events[l[i]].push_back({i,v[i]});
        events[r[i]+1].push_back({i,-v[i]});
    }
    
    vector<int>s(n);
    tree_init();
    
    for(int i=0;i<n;i++){
        for(auto event:events[i])
            tree_add(event.first,event.second);
        s[i]=tree_get(c[i]);
    }
    
    return s;
}

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