Submission #535752

#	Time	Username	Problem	Language	Result	Execution time	Memory
535752		MurotY	Distributing Candies (IOI21_candies)	C++17	11 / 100	1682 ms	11464 KiB

candies

#include "candies.h"
#include <bits/stdc++.h>
#include <vector>
#define ll long long
using namespace std;
vector <ll> tree;
ll sz=1, ans;
bool ok=0;
void init(ll n){
    while (sz < n) sz*=2;
    tree.assign(2*sz, 0);
    return ;
}

void add(ll l, ll r, ll v, ll x, ll lx, ll rx){

    if (lx >= r || rx <= l) return ;
    if (l <= lx && rx <= r) {
        tree[x]+=v;
        return ;
    }
    ll m=(lx+rx)/2;
    add(l, r, v, 2*x+1, lx, m);
    add(l, r, v, 2*x+2, m, rx);
    return ;
}
void help(ll i, ll x, ll lx, ll rx, ll res=0){

    if (rx-lx == 1){
        if (!ok) ans=res+tree[x];
        ok=1;
        return;
    }
    ll m=(lx+rx)/2;
    ll x1=x;
    if (i < m) help(i, 2*x+1, lx, m, res+tree[x1]);
    else help(i, 2*x+2, m, rx, res+tree[x1]);
    return ;
}
std::vector<int> distribute_candies(std::vector<int> c, std::vector<int> l,
                                    std::vector<int> r, std::vector<int> v) {
    int n = c.size();
    init(n);
    vector<int> s(n);
    if (n <= 2000){
        for (int i=0;i<l.size();i++){
            //    cout << v[i] <<"\n";
            for (int j=l[i];j<=r[i];j++) s[j]=min(c[j], max(0, s[j]+v[i]));
        }
        return s;
    }
    for (int i=0;i<l.size();i++){
        add(l[i], r[i]+1, v[i], 0, 0, sz);
    }
    for (int i=0;i<n;i++){
        ok=0;
        ans=0;
        help(i, 0, 0, sz);
        if (ans > c[i]) s[i]=c[i];
        else s[i]=ans;
    }
    return s;
}

Compilation message (stderr)

candies.cpp: In function 'std::vector<int> distribute_candies(std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>)':
candies.cpp:46:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   46 |         for (int i=0;i<l.size();i++){
      |                      ~^~~~~~~~~
candies.cpp:52:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   52 |     for (int i=0;i<l.size();i++){
      |                  ~^~~~~~~~~

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