[BZOJ4476] [JSOI2015] 送礼物 (01分数规划+ST表)

[BZOJ4476] [JSOI2015] 送礼物 (01分数规划+ST表)

题面

给出n,k,l,r和序列a，要求从a中选一段连续的区间[i,j]出来，使得M(i,j)-m(i,j)/(j-i+k)最大，且[i,j]长度在区间[l,r]内。
M(i,j)表示[i,j]中的最大值，m(i,j)表示[i,j]中的最小值。
Ai< =10^8,N,K< = 50,000

分析

看到\(\frac{max(i,j)-min(i,j)}{j-i+k}\)的形式，我们敏感的想到01分数规划。考虑最终答案我们取的最大值和最小值在序列中的位置。

1.如果最大值和最小值在序列中的距离<L,那么用长度为L的区间包含它最优。因为如果区间更长的话，分子不变，分母会变大，不优。这种情况区间长度已经确定，可以不用01分数规划，直接用单调队列\(O(n)\)求即可。类似滑动窗口问题。

2.如果最大值和最小值在序列中的距离>=L,那么它们在区间的左右端点最优。同样如果区间更长的话，分子不变，分母会变大，不优。

按照01分数规划的套路，二分答案mid，判断当前答案是比mid大还是比mid小。如果比mid大，则有
\(|a_i-a_j| \geq mid(j-i+k)\)

(1)$ (a[i]+mid \times i)-(a[j]+mid \times j) \geq mid\times k $,

(区间左端点为最大值)只要暴力枚举i,再在i右侧满足区间长度条件的范围[i+L-1,i+R-1]内求一个\(a[j]+mid \times j\)最小的j即可

(2)$ (a_j-mid \times j)-(a_i-mid \times i) \geq mid \times k ,$

(区间右端点为最大值)只要暴力枚举j,再在j左侧满足区间长度条件的范围[i-L+1,i-R+1]内求一个\(a[j]+mid \times j\)最小的j即可

如果[i,j]内有比a[i]更大的a[k]怎么办？枚举到k的时候就会计入答案，且k为左端点的答案肯定比i优，会覆盖掉i为左端点的不合法情况
用ST表维护最小值即可。

时间复杂度\(O(n \log ^2 n)\),因为每次检查mid都要重新建一次ST表

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<cmath> 
#include<deque>
#define INF 1e9
#define eps 1e-6 
#define maxn 50000
#define maxlogn 20
using namespace std;
inline void qread(int &x){
    x=0;
    int sign=1;
    char c=getchar();
    while(c<'0'||c>'9'){
        if(c=='-') sign=-1;
        c=getchar();
    }
    while(c>='0'&&c<='9'){
        x=x*10+c-'0';
        c=getchar();
    }
    x=x*sign; 
}
int t;
int n,K,L,R;
int a[maxn+5];


//最大值最小值在两端的情况 
//最大值最小值在区间两端的时候显然更优 
// 01分数规划的套路，二分答案mid
//|a[i]-a[j]| >=mid(j-i+k)
//(1) (a[i]+mid*i)-(a[j]+mid*j)>=mid*k , 只要暴力枚举i,再在i右侧满足区间长度条件的范围内求一个值最小的j
//(2) (a[j]-mid*j)-(a[i]+mid*i)>=mid*k , 只要暴力枚举j,再在j左侧满足区间长度条件的范围内求一个值最小的j
//如果[i,j]内有比a[i]更大的a[k]怎么办？枚举到k的时候就会计入答案，且k为左端点的答案肯定比i优，会覆盖掉i为左端点的不合法情况 
int log_2[maxn+5]; 
double st1[maxlogn+5][maxn+5];//维护a[i]+mid*i的最小值 
double st2[maxlogn+5][maxn+5];//维护a[i]-mid*i的最小值 
inline double max(double a,double b){
    return a>b?a:b;
}
inline double min(double a,double b){
    return a<b?a:b;
}
double query1(int l,int r){
    int k=log_2[r-l+1];
    return min(st1[k][l],st1[k][r-(1<<k)+1]);
} 
double query2(int l,int r){
    int k=log_2[r-l+1];
    return min(st2[k][l],st2[k][r-(1<<k)+1]);
} 
bool check(double mid){
    for(int i=1;i<=n;i++) st1[0][i]=a[i]+mid*i;
    for(int i=1;i<=n;i++) st2[0][i]=a[i]-mid*i;
    for(int j=1;(1<<j)<=n;j++){
        for(int i=1;i+(1<<j)-1<=n;i++){
            st1[j][i]=min(st1[j-1][i],st1[j-1][i+(1<<(j-1))]);
            st2[j][i]=min(st2[j-1][i],st2[j-1][i+(1<<(j-1))]);
        }
    } 
    double ans=-INF;
    for(int i=1;i<=n;i++){
        if(i+L-1<=n) ans=max(ans,a[i]+mid*i-query1(i+L-1,min(i+R-1,n))); 
    }
    for(int i=1;i<=n;i++){//这里的i对应的是上面的j 
        if(i-L+1>=1) ans=max(ans,a[i]-mid*i-query2(max(i-R+1,1),i-L+1));
    }
    return ans>=mid*K;
} 

//区间长度为L的时候，区间最大值不一定在左右端点，也可能成为最大答案 
//直接长度为L的滑块窗口求解 
double sp_solve(){
    deque<int>qmax;
    deque<int>qmin;
    double ans=-INF;
    for(int i=1;i<=n;i++){
        while(!qmax.empty()&&qmax.front()<i-L+1) qmax.pop_front(); 
        while(!qmin.empty()&&qmin.front()<i-L+1) qmin.pop_front(); 
        while(!qmax.empty()&&a[qmax.back()]<a[i]) qmax.pop_back();
        qmax.push_back(i);
        while(!qmin.empty()&&a[qmin.back()]>a[i]) qmin.pop_back();
        qmin.push_back(i);
        ans=max(ans,1.0*(a[qmax.front()]-a[qmin.front()])/(L-1+K));
    }
    return ans;
}
double bin_search(){
    double l=0,r=1000;
    double mid;
    double res=0; 
//  printf("db %d\n",check(0+eps)?1:0);
    while(r-l>=eps){
        mid=(l+r)/2;
        if(check(mid)){
            res=mid;
            l=mid+eps;
        }
        else r=mid-eps;
    }
    return res;
}
int main(){
//  freopen("1.in","r",stdin);
    log_2[0]=-1;
    for(int i=1;i<=maxn;i++) log_2[i]=log_2[i>>1]+1;
    qread(t);
    while(t--){
        qread(n);
        qread(K);
        qread(L);
        qread(R);
        for(int i=1;i<=n;i++) qread(a[i]);
        double ans1=sp_solve();
//      printf("ans1=%.4f\n",ans1);
        double ans2=bin_search();
        double ans=max(ans1,ans2); 
        printf("%.4f\n",ans);
    }

} 

转载于:https://www.cnblogs.com/birchtree/p/11360244.html