首页 > 代码库 > BZOJ 2600 IOI 2011 ricehub 二分

BZOJ 2600 IOI 2011 ricehub 二分

题目大意:给出粮食的位置,给出初始的钱数,问粮仓设在哪里会让收到的粮食最多。粮食运到粮仓需要花费|x - x‘|的费用。


思路:看到L和B的范围都大的惊人,只能从R来下手。能不能从每个粮食的位置入手来搞点什么。

首先要想收购的尽量多,收购的粮食应该是一段连续的区间。对于每个粮食的位置,以这个粮食的位置为起点所能够收购到的粮食是单调的。这个就可以二分来确定了。一段区间内到达所有点最近的点是中位数,所以就可以O(1)判断了。

(这真的是IOI题么。。


CODE:

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define MAX 100010
using namespace std;

int cnt,range;
long long money;
long long src[MAX],sum[MAX];

inline long long Calc(int l,int r)
{
	int mid = (l + r) >> 1;
	long long pos = src[mid];
	return pos * (mid - l) - (sum[mid - 1] - sum[l - 1]) + sum[r] - sum[mid] - pos * (r - mid);
}

int main()
{
	cin >> cnt >> range >> money;
	for(int i = 1; i <= cnt; ++i) {
		scanf("%lld",&src[i]);
		sum[i] = sum[i - 1] + src[i];
	}
	int _max = 0;
	for(int i = 1; i <= cnt; ++i) {
		int l = 1,r = cnt - i + 1,ans = 0;
		while(l <= r) {
			int mid = (l + r) >> 1;
			if(Calc(i,i + mid - 1) <= money)
				l = mid + 1,ans = mid;
			else	r = mid - 1;
		}
		_max = max(_max,ans);
	}
	cout << _max << endl;
	return 0;
}


BZOJ 2600 IOI 2011 ricehub 二分