Submission #466511

#	Time	Username	Problem	Language	Result	Execution time	Memory
466511		JvThunder	Distributing Candies (IOI21_candies)	C++17	29 / 100	146 ms	16828 KiB

candies

#include <bits/stdc++.h>
typedef long long ll;
using namespace std;
 
const int MAXQ = 1000005;
const ll INF = (ll)(1e18);

vector<int> distribute_candies(vector<int> c, vector<int> L, vector<int> R, vector<int> v) 
{
    int n = c.size(); int q = v.size();
    
    vector<ll> p;
    p.push_back(INF); p.push_back(0LL);
    for(int i=0;i<q;i++) p.push_back(p.back() + v[i]);
    q+=2;

    vector<ll> smax(q),smin(q);
    smax[q-1] = smin[q-1] = p[q-1];
    for(int i=q-2;i>=0;i--)
    {
        smax[i] = max(smax[i+1],p[i]);
        smin[i] = min(smin[i+1],p[i]);
    }

    vector<int> final_A(n);
    for(int i=0;i<n;i++)
    {
        int l = 0; int r = q-1;
        int ret = -1;
        while(l<=r)
        {
            int mid = (l+r)/2;
            if(smax[mid]-smin[mid]>=c[i])
            {
                ret = mid;
                l = mid+1;
            }
            else r = mid-1;
        }
        if(p[ret]==smax[ret]) final_A[i] = p[q-1]-smin[ret];
        else final_A[i] = c[i]-(smax[ret]-p[q-1]);
    }
    return final_A;
}

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