二分+前缀和：[NOIP2011]聪明的质监员

题目

聪明的质检员

思路

二分

• 二分应用的条件为：满足单调性或可以舍去局部。对于此题来说单调性十分明显随着W的增大，Y不断减小。利用二分法，找到最接近S的两个值（一个位于S左侧，一个位于右侧），然后选取S-Y的绝对值最小的即可。

前缀和

对于某个W的值，如果使用暴力求解Y，复杂度为O(mn)，那么总的复杂度为O(mnlog(Wmax))这样显然会TLE。

• 观察题目中让求的式子，我们可以利用前缀和，求前缀和只需要O(n)的复杂度，对于[l,r]区间查询只需要prefsum[r]-prefsum[l-r]，这样总的复杂度变为了O((m+n)log(Wmax))

AC代码

#include<bits/stdc++.h>
#define N 200001
#define ll long long int
using namespace std;
struct good{
	int w;
	int v;
}G[N];
struct qujian{
	int L;
	int R;
}Q[N];
int n,m;
ll S;
//计算,利用前缀和
ll compute(int x){
	ll prefsum[n+1];
	ll prefnum[n+1]; 
	memset(prefsum,0,sizeof(prefsum));
	memset(prefnum,0,sizeof(prefnum));
	//计算前缀和 
	for(int i=1;i<=n;i++){
		prefsum[i] = prefsum[i-1]+(G[i].w>=x?G[i].v:0);
		prefnum[i] = prefnum[i-1]+(G[i].w>=x?1:0);
	}
	ll result=0;
	for(int i=1;i<=m;i++){
		result+=(prefnum[Q[i].R]-prefnum[Q[i].L-1])*(prefsum[Q[i].R]-prefsum[Q[i].L-1]); 
	}
	return result;
}
int main(){
	cin>>n>>m>>S;
	ll ans = 0;
	int maxW = 0;
	for(int i=1;i<=n;i++){
		cin>>G[i].w>>G[i].v;
		maxW=max(maxW,G[i].w);
	}
	for(int i=1;i<=m;i++)	cin>>Q[i].L>>Q[i].R;
	int l = 1;
	int r = maxW+1;
	int mid;
	ll temp;
	//二分 
	while(l<r){
		mid = (l+r)/2;
		temp = compute(mid);
		if(temp==S)
			break;
		else if(temp>S){
			l = mid;
		}
		else{
			r = mid;
		}
		if(l==r-1){
			ans = min(abs(compute(l)-S),abs(compute(r)-S));
			break;
		} 
	}
	cout<<ans;
	return 0;
}