【JZOJ4772】运输妹子

Description

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

Solution

首先，选择的农场肯定是一个连续的区间。

其次，选择一个区间，一定是在最中间的农场建仓库。

中点为什么不行？（开始我就是这么打的）因为会超出一些匹配的范围。

那么设两个指针 l,r ，当区间 [l,r] 合法时就将 r+1 ，否则就将 l+1 。

二分也行，当初我也是这样打的。

Code

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
#define N 100001
#define inf 3000000000000000ll
#define ll long long
using namespace std;
ll a[N],s[N];
ll L,w;
int n;
int main()
{
    cin>>n>>L>>w;
    fo(i,1,n) scanf("%lld",&a[i]);
    sort(a+1,a+n+1);
    fo(i,1,n) s[i]=s[i-1]+a[i];
    int l=1,r=1,c=0;
    ll mid,s1=0,s2,n1=0,n2;
    ll tot;
    int ans=0;
    while(r<=n)
    {
        mid=a[(l+r)/2];
        while(a[c+1]<=mid) c++,s1+=a[c],n1++;
        s2=s[r]-s[l-1]-s1;
        n2=r-l+1-n1;
        tot=n1*mid-s1+s2-n2*mid;
        if(tot<=w)
        {
            if(ans<r-l+1) ans=r-l+1;
            r++;
        }
        else
        {
            s1-=a[l];
            n1--;
            l++;r++;
        }
    }
    cout<<ans;
}