洛谷 聪明的质检员

原题 P1314

题目描述

小T 是一名质量监督员，最近负责检验一批矿产的质量。这批矿产共有 n 个矿石，从 1 到 n 逐一编号，每个矿石都有自己的重量 wi​ 以及价值 vi​ 。检验矿产的流程是：

1. 给定m 个区间 [li​,ri​]；
2. 选出一个参数 W；
3. 对于一个区间[li​,ri​]，计算矿石在这个区间上的检验值 yi​

其中 j 为矿石编号。

这批矿产的检验结果 y 为各个区间的检验值之和。即：

若这批矿产的检验结果与所给标准值 s 相差太多，就需要再去检验另一批矿产。小T 不想费时间去检验另一批矿产，所以他想通过调整参数 W 的值，让检验结果尽可能的靠近标准值 s，即使得 ∣s−y∣ 最小。请你帮忙求出这个最小值。

输入格式

第一行包含三个整数 n,m,s，分别表示矿石的个数、区间的个数和标准值。

接下来的 n 行，每行两个整数，中间用空格隔开，第i+1 行表示 i 号矿石的重量 wi​ 和价值 vi​。

接下来的 m 行，表示区间，每行两个整数，中间用空格隔开，第i+n+1 行表示区间 [li​,ri​] 的两个端点 li​ 和 ri​。注意：不同区间可能重合或相互重叠。

输出格式

一个整数，表示所求的最小值。

输入输出样例

输入 #1

5 3 15 
1 5 
2 5 
3 5 
4 5 
5 5 
1 5 
2 4 
3 3 

输出 #1

10

说明/提示

解题思路

前缀和+二分。

具体实现呢，我们首先找出W的范围，然后利用二分法寻找，时间复杂度为（logN）。

这一步做完后，又有新问题了，我们如何确定W缩小还是增大呢？

我们可以先利用前缀和思想，把我们的总w,v得到，具体怎么做看代码，然后将它们相乘，最后Y>s缩小，反之扩大，这一步的时间复杂度为O(n).

所以总时间复杂度为O(nlogN)（N与n不同）。

注意：Y=0时，W等于max+1。

AC代码

#include <iostream>
#include <algorithm>
#include <cmath>
using namespace std;
int n,m;
const int N=2e5+5;
int w[N],v[N],r[N],l[N];
long long ans,sum,Y,pre_w[N],pre_v[N],s;
bool check(int W){
	Y=0;sum=0; 
	fill(pre_w+1,pre_w+N,0);
	fill(pre_v+1,pre_v+N,0);
	for(int i=1;i<=n;i++)
		if(w[i]<W) pre_w[i]=pre_w[i-1],pre_v[i]=pre_v[i-1];     //小于W,舍弃
		else pre_w[i]=pre_w[i-1]+1,pre_v[i]=pre_v[i-1]+v[i];    //反之+
	for(int i=1;i<=m;i++) {
		Y+=(pre_w[r[i]]-pre_w[l[i]-1])*(pre_v[r[i]]-pre_v[l[i]-1]); //注意这里是r-l-1
	}
	sum=llabs(Y-s);
	return Y>s;
}
int main(){
	scanf("%d %d %lld",&n,&m,&s);
	int Min=0x7fffffff,Max=0;
	for(int i=1;i<=n;i++){
		scanf("%d %d",w+i,v+i);
		Min=min(Min,w[i]);
		Max=max(Max,w[i]);
	}
	
	for(int i=1;i<=m;i++){
		scanf("%d %d",l+i,r+i);
		 
	}
	int left=Min,right=Max+1;     //可能Y=0
	ans=0x3f3f3f3f3f3f3f3f;
	while(left<right){
		int mid=(right+left)>>1;
		if(check(mid)) left=mid+1;	
		else right=mid;
		if(ans>sum) ans=sum;
	}
	printf("%lld",ans);
	return 0;
}