做题思路(超时解法):根据题意,要求S-Y的最小绝对值,因为参数W未知,所以最简单的想法是枚举W,每枚举一个W,先计算出Y,然后将S-Y的绝对值与现有答案进行比较,取小的那个即可。做题时,发现如果依次枚举W,程序运行会很慢,于是发现可以利用二分猜答案猜W,每猜一个W,如果计算的Y大于S,则说明区间里取的矿石多了,W可以猜大点;如果计算的Y小于S,则说明区间里取的矿石少了,W可以猜小点。但在计算Y时,考试时想得比较简单,直接先枚举每个区间,在枚举每个区间里的矿石,从而求得Y,结果最后发现程序依然超时,原因就是没有优化求Y的方法。这种算法的时间复杂度为O(k*M*N)(k为猜答案的次数)。
解题思路(正解):在计算Y时,因为是从区间里选矿石,所以可以在枚举区间时,先进行预处理,即求出前i个矿石的个数和价值和(前缀和),之后,在枚举第j个区间时(设第j个区间右端点为R,左端点为L),Yj=(前R个矿石个数-前L-1个矿石个数(注意是L-1,因为第L个矿石也在区间里))*(前R个矿石价值和-前L-1个矿石的价值和),最后Y即为Yj之和。(其余思路与做题思路一样)该算法的时间复杂度为O(k*max(N,M))(k为猜答案的次数)。
#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int maxn=200005;
typedef long long LL;
int N,M;
LL S,sum,ans=40000000000000001ll;
struct data1 //记录矿石
{
int w,v;
};
data1 a[maxn];
struct data2 //记录区间
{
int l,r;
};
data2 b[maxn];
LL t1[maxn],t2[maxn];
bool check(int m)
{
LL sum=0; //计算Y
t1[0]=t2[0]=0;
for(int i=1;i<=N;i++) //进行预处理
{
t1[i]=t1[i-1],t2[i]=t2[i-1];
if(a[i].w>=m) t1[i]++,t2[i]+=a[i].v;
}
for(int i=1;i<=M;i++)
{
sum+=(t1[b[i].r]-t1[b[i].l-1])*(t2[b[i].r]-t2[b[i].l-1]);
}
ans=min(ans,abs(S-sum)); //每次计算出Y,对答案进行更新
if(sum>=S) return 1;
else return 0;
}
int main()
{
freopen("qc.in","r",stdin);
freopen("qc.out","w",stdout);
scanf("%d%d",&N,&M);
cin>>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 A=0,B=1000000;
for(int i=0;i<30;i++) //二分猜参数W
{
int m=(A+B)/2;
if(check(m)) A=m+1;
else B=m-1;
}
cout<<ans<<'\n';
return 0;
}
考后反思:有时做题的代码优化不止一处,对于可能超时的地方(主要是枚举)要仔细想想是否可以进行预处理或有高效算法进行优化。