【JZOJ 4772】 运输妹子

Description

小轩轩是一位非同一般的的大农(lao)场(si)主(ji)，他有一大片非同一般的农田，并且坐落在一条公路旁（可以认为是数轴），在他的农田里种的东西也非同一般——不是什么水稻小麦，而是妹子。
在小轩轩的细心培育下，他的大片农田都要结出妹子啦！但是他的农田分布实在是太广阔了，他担心自己的妹子会令路过的人想入非非，于是他想要把所有农田上的妹子都集中到一个仓库里面，贮存起来。可是妹子太多，他叫来了一辆卡车，这辆卡车刚好可以装满一个农田的妹子，并且在满载的情况下，运满满一卡车妹子走1米的费用是1元。由于小轩轩技术精湛，他的每个农田产量都是一样的。即把一个农田的妹子都运到仓库费用为农田与仓库坐标差值的绝对值。理想很美好，但现实很残酷——小轩轩还没有想好在什么位置搭建他的仓库，而且他的运输费用是有限的。
请你帮忙计算一下，在什么位置搭建仓库，使得小轩轩能收获的妹子最多。（仓库的位置可与农田的位置重合，并且在公路长度范围内）。
对于 100% 的数据，N≤100000，L≤1000000000，W≤2000000000000000。

Analysis

首先有一个显然的性质，仓库必然在某个点上。
还有一个性质：一段区间仓库必然建在最中间的点上。
我们可以维护两个指针，包含一段区间，然后根据前缀和判断指针移动即可。

Code

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int N=100010;
int n,a[N];
ll b[N];
int main()
{
    int n,m;
    ll ans=0,lim;
    scanf("%d %d %lld",&n,&m,&lim);
    fo(i,1,n) scanf("%d",&a[i]);
    sort(a+1,a+n+1);
    fo(i,1,n) b[i]=b[i-1]+a[i];
    ll l=1,r=2,mid;
    while(l<r)
    {
        while(r<=n)
        {
            mid=(l+r)>>1;
            ll t=(mid-l)*a[mid]-b[mid-1]+b[l-1]+b[r]-b[mid]-(r-mid)*a[mid];
            if(t<=lim) r++;
            else break;
        }
        ans=max(ans,r-l);
        l++;
    }
    printf("%lld",ans);
    return 0;
}