洛谷1314 聪明的质监员(二分)(前缀和)

版权声明:本文为博主原创文章,未经博主允许不得转载,除非先点了赞。 https://blog.csdn.net/A_Bright_CH/article/details/83080581

题目

洛谷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;
}

猜你喜欢

转载自blog.csdn.net/A_Bright_CH/article/details/83080581