题目
思路
二分
- 二分应用的条件为:满足单调性或可以舍去局部。对于此题来说单调性十分明显随着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;
}