Submission #588400

#	Time	Username	Problem	Language	Result	Execution time	Memory
588400		Blagojce	Distributing Candies (IOI21_candies)	C++17	Compilation error	0 ms	0 KiB

candies

#include "candies.h"

#include <bits/stdc++.h>
#define fr(i, n, m) for(int i = (n); i < (m); i ++)
#define pb push_back
#define st first
#define nd second
#define all(x) begin(x), end(x)
#define pq priority_queue


using namespace std;

typedef long long ll;


int n;
int q;

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 = v.size();
	
	ll sum[q+1];
	sum[0] = 0;
	
	fr(i, 1, q+1){
		sum[i] = sum[i-1] + v[i-1];
	}
	
	
	
	
	
	
	
    ll suf[q+1];
    suf[q] = 0;
    for(int i = q-1; i >= 0; i --){
		suf[i] = suf[i+1] + v[i];
	}
	
	
	ll Min[q+2];
	ll Max[q+2];
	Min[q+1] = 1e18;
	Max[q+1] =-1e18;
	for(int i = q; i >= 0; i--){
		Min[i] = min(Min[i+1], sum[i]);
		Max[i] = max(Max[i+1], sum[i]);
	}
	
	
	vector<ll> sol;
	fr(i, 0, n){
		int k1 = q+1;
		for(int b = (q+3)/2; b >= 1; b /= 2){
			while(k1-b >= 0 && Max[k1-b] - Min[k1-b] <= c[i]) k1 -= b;
		}
		
		ll h = sum[q];
		if(k1 == 0){
			sol.pb(h);
			continue;
		}
		
		
		if(sum[k1] > sum[k1-1]){
			sol.pb(h-Max[k1]+c[i]);
		}
		else{
			sol.pb(h-Min[k1]);
		}
		
	}
    return sol;
}
/*

int main(){
	vector<int> ans = distribute_candies({10, 15, 13}, {0, 0}, {2, 2}, {20, -11});
	for(auto u : ans) cout<< u<<' ';
	cout<<endl;
	
}
*/

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:76:12: error: could not convert 'sol' from 'vector<long long int>' to 'vector<int>'
   76 |     return sol;
      |            ^~~
      |            |
      |            vector<long long int>