Submission #677882

#	Time	Username	Problem	Language	Result	Execution time	Memory
677882		APROHACK	사탕 분배 (IOI21_candies)	C++17	11 / 100	110 ms	16292 KiB

candies

#include "candies.h"
#include <bits/stdc++.h>
#define ll long long
#define pb push_back
#define ff first
#define ss second
using namespace std;
int n, q;
vector<int>s;
void initialize(){
    for (int i = 0; i < n; i++){
        s.pb(0);
    }
}


void case1(vector<int> c, vector<int> l, vector<int> r, vector<int> v){
    for(int i = 0 ; i < q ; i  ++){
            for(int j = l[i] ; j <= r[i] ; j  ++){
                if(v[i] > 0){
                    s[j] = min(c[j], v[i]+s[j]);
                    //cout << "s[" << j << "] = " << s[j] << endl; 
                }else{
                    s[j] = max(0, s[j]+v[i] );
                    // << "s[" << j << "] = " << s[j] << endl;
                }
            }
        }
}

void case2(vector<int> c, vector<int> l, vector<int> r, vector<int> v){
    ll prefix[n+2];
    memset(prefix, 0, sizeof prefix) ;
    for(int i =0 ; i < q ; i ++){
        prefix[l[i]]+=v[i];
        prefix[r[i]+1]-= v[i];
    }
    ll cuenta =0 ;
    for(int i = 0 ; i <n ; i ++){
        cuenta+=prefix[i];
        //cuenta = min((ll)1000000000, cuenta);
        //cout << "cuenta = " << cuenta << endl;
        s[i] = min(cuenta, (ll)c[i]);
    }

}
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();
    initialize();
    if(n<=2000 and q <= 2000){
        case1(c, l, r, v);
    }
    else{
        case2(c, l, r, v);
    }
    

    return s;
}

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