洛谷1314 聪明的质监员（二分）（前缀和）

题目

 

洛谷1314 聪明的质监员

题解

二分+前缀和
很显然满足二分性，随着W的增大，Y会变小，我们要找的是一个abs(S-Y)最小的值。
一开始，我想着把abs拆开来看，讨论min(W-Y)和min(Y-W)。后来发现一个更牛逼的做法，只要一次二分，因为是要与S做绝对值，所以我们把Y值不断逼近S，有点像二分查找。
至于如何快速求出对于当前W的Y呢？我们用前缀和。对于每个W都做一次前缀和，求一个sum记录v的前缀和，和一个cnt记录1~i中有几个w是大于W的。这样在O（n）的预处理后，对于每段区间就可以O（1）求出Yi。

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=2e5+10;
int n,m;ll s;

struct U{int w,v,cnt;}a[maxn];
struct O{int l,r;}b[maxn];

ll sum[maxn];int cnt[maxn];
ll calc(int W)
{
    ll re=0;
    for(int i=1;i<=n;i++)
        if(a[i].w>=W) sum[i]=sum[i-1]+a[i].v,cnt[i]=cnt[i-1]+1;
        else sum[i]=sum[i-1],cnt[i]=cnt[i-1];
    for(int i=1;i<=m;i++)
        re+=(cnt[b[i].r]-cnt[b[i].l-1])*(sum[b[i].r]-sum[b[i].l-1]);
    return re;
}

int main()
{
    scanf("%d%d%lld",&n,&m,&s);
    for(int i=1;i<=n;i++) scanf("%d%d",&a[i].w,&a[i].v);
    for(int i=1;i<=m;i++) scanf("%d%d",&b[i].l,&b[i].r);
    int l=1,r=1e6;ll ans;
    while(l<=r)
    {
        int mid=l+r>>1;
        ll Y=calc(mid);ans=min(ans,abs(Y-s));
        if(Y<=s) r=mid-1;
        else l=mid+1;
    }
    printf("%lld\n",ans);
    return 0;
}