Submission #418382

#	Time	Username	Problem	Language	Result	Execution time	Memory
418382		ismoilov	Rice Hub (IOI11_ricehub)	C++14	100 / 100	16 ms	2164 KiB

ricehub

#include "ricehub.h"
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define IOS ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
#define all(x) (x).begin(), (x).end()
#define fp(a,i,c) for(int (a) = (i); (a) < (c); (a)++)
#define fpp(a,i,c) for(int (a) = (i); (a) <= (c); (a)++)
#define fm(a,i,c) for(int (a) = (i); (a) > (c); (a)--)
#define fmm(a,i,c) for(int (a) = (i); (a) >= (c); (a)--)
deque <int> fir, sec;
ll sum_fir, sum_sec;

ll check(){
	int ans;
	if(sec.size() == fir.size())
		ans = sec.front();
	else
		ans = fir.back();
	ll c = sum_sec - sum_fir;
	c += (ll) ans * fir.size();
	c -= (ll) ans * sec.size();
	return c;
}

int besthub(int r, int l, int X[], ll b){
	int ans = 0;
	fp(i,0,r){
		sec.push_back(X[i]);
		sum_sec += X[i];
		if(fir.size() < sec.size()){
			sum_sec -= sec.front();
			sum_fir += sec.front();
			fir.push_back(sec.front());
			sec.pop_front();
		}
		
		while(b < check()){
			if(fir.empty()){
				sum_sec -= sec.back();
				sec.pop_back();
			}
			else{
				sum_fir -= fir.front();
				fir.pop_front();
			}
			
			if(fir.size() < sec.size()){
				sum_sec -= sec.front();
				sum_fir += sec.front();
				fir.push_back(sec.front());
				sec.pop_front();
			}
		}
		ans = max(ans, (int) fir.size() + (int) sec.size());
	}
	return ans;
}

Compilation message (stderr)

ricehub.cpp: In function 'int besthub(int, int, int*, ll)':
ricehub.cpp:7:27: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
    7 | #define fp(a,i,c) for(int (a) = (i); (a) < (c); (a)++)
      |                           ^
ricehub.cpp:28:2: note: in expansion of macro 'fp'
   28 |  fp(i,0,r){
      |  ^~

• Subtask 1: 17 / 17
• Subtask 2: 25 / 25
• Subtask 3: 26 / 26
• Subtask 4: 32 / 32