Description
小轩轩是一位非同一般的的大农(lao)场(si)主(ji),他有一大片非同一般的农田,并且坐落在一条公路旁(可以认为是数轴),在他的农田里种的东西也非同一般——不是什么水稻小麦,而是妹子。
在小轩轩的细心培育下,他的大片农田都要结出妹子啦!但是他的农田分布实在是太广阔了,他担心自己的妹子会令路过的人想入非非,于是他想要把所有农田上的妹子都集中到一个仓库里面,贮存起来。可是妹子太多,他叫来了一辆卡车,这辆卡车刚好可以装满一个农田的妹子,并且在满载的情况下,运满满一卡车妹子走1米的费用是1元。由于小轩轩技术精湛,他的每个农田产量都是一样的。即把一个农田的妹子都运到仓库费用为农田与仓库坐标差值的绝对值。理想很美好,但现实很残酷——小轩轩还没有想好在什么位置搭建他的仓库,而且他的运输费用是有限的。
请你帮忙计算一下,在什么位置搭建仓库,使得小轩轩能收获的妹子最多。(仓库的位置可与农田的位置重合,并且在公路长度范围内)。
Input
第一行三个正整数 N,L,W,分别表示农田个数、公路总长度、总钱数。
接下来 N 行,每一行描述一个农田的坐标。(可能有多个农田位于同一位置)
Output
一个整数,小轩轩最多能够收藏多少个农田的妹子。
Sample Input
5 23 18
4
6
14
18
22
Sample Output
3
Data Constraint
对于 30% 的数据,N≤1000,L≤10000。
对于 60% 的数据,N≤10000,L≤1000000。
对于 100% 的数据,N≤100000,L≤1000000000,W≤2000000000000000。
The Solution
此题可用二分。
吐槽一句: L压根就没用 (#‵′)凸
我们可以二分答案,
然后枚举区间。
在这个区间里,所有农场到仓库的花费要<=w即为合法。
而仓库的位置很显然是这段区间里的中位数,即最中间的农田。
然而确定了农田,花费怎么算呢?
我们有很多种计算方法。
我用的是一个前缀和数组Pre,记录前i个坐标的和。
我们可以拿样例说明
4 6 14 18 22
p: 4 10 24 42 64
假设我们选定1~4区间,
那么mid=2
即仓库为6这个坐标。
花费为:
18−6+14−6+6−4
=18+14−2∗6+6∗1−4
然后计算公式就可以化简了
推广为一般式为:
Pre[y]−Pre[mid]−(y−mid)∗a[mid]+a[mid]∗(mid−x)−(Pre[mid]−a[mid]−Pre[x−1])
这里ps一下
i~j区间的和为Pre[j]-Pre[i-1];
那么i+1~j区间的和为Pre[j]-Pre[i];
最后二分完要特判一下
CODE
#include <cstdio>
#include <iostream>
#include <cmath>
#include <algorithm>
#include <cstring>
#define fo(i,a,b) for (int i=a;i<=b;i++)
#define N 100005
using namespace std;
int n,len;
long long a[N],Pre[N],w;
inline long long read()
{
long long x=0,w=1;
char ch=getchar();
while(ch>'9'||ch<'0'){if(ch=='-')w=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*w;
}
long long Calc(int x,int y)
{
int mid = (x + y) >> 1;
long long res=Pre[y] - Pre[mid] - (y-mid) * a[mid] + a[mid] * (mid - x) - (Pre[mid] - a[mid] - Pre[x - 1]);
return res;
}
bool Check(int mid)
{
fo(i,1,n - mid + 1)
if (Calc(i,i+mid-1)<=w) return true;
return false;
}
int main()
{
freopen("data.in","r",stdin);
scanf("%d%d",&n,&len);w=read();
fo(i,1,n) a[i]=read();
sort(a+1,a+1+n);
fo(i,1,n) Pre[i] = Pre[i-1] + a[i];
int l = 0,r = n;
while (l < r - 1)
{
int mid = (l + r) >> 1;
if (Check(mid)) l = mid;
else r = mid - 1;
}
if (Check(r)) printf("%d\n",r);
else printf("%d\n",l);
}