Submission #1022567

#	Time	Username	Problem	Language	Result	Execution time	Memory
1022567		socpite	Rice Hub (IOI11_ricehub)	C++14	100 / 100	163 ms	3056 KiB

ricehub

#include "ricehub.h"
#include<bits/stdc++.h>
using namespace std;

const int maxn = 1e5+5;
const int INF = 1e9+5;

int A[maxn];
long long pfx[maxn];


int besthub(int R, int L, int X[], long long B)
{
	int ans = 0;
	for(int i = 1; i <= R; i++){
		A[i] = X[i-1];
		pfx[i] = pfx[i-1] + A[i];
	}
	for(int i = 1; i <= R; i++){
		int l = 0, r = INF;
		while(l < r){
			int mid = (l+r+1)>>1;

			long long sum = 0;
			int pl = lower_bound(A+1, A+R+1, A[i] - mid) - A - 1;
			sum += 1LL*A[i]*(i - pl) - (pfx[i] - pfx[pl]);

			int pr = upper_bound(A+1, A+R+1, A[i]+mid) - A - 1;
			sum += (pfx[pr] - pfx[i-1]) - 1LL*A[i]*(pr - (i-1));
			if(sum <= B)l = mid;
			else r = mid-1;
		}
		if(l == INF)return R;
		int mid = l;

		long long sum = 0;
		int pl = lower_bound(A+1, A+R+1, A[i] - mid) - A - 1;
		sum += 1LL*A[i]*(i - pl) - (pfx[i] - pfx[pl]);

		int pr = upper_bound(A+1, A+R+1, A[i]+mid) - A - 1;
		sum += (pfx[pr] - pfx[i-1]) - 1LL*A[i]*(pr - (i-1));
		
		ans = max<long long>(ans, pr - pl + (B - sum)/(l+1));
	}
	return ans;
}

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