UVA_12697 满足条件的最短连续和 线段树维护

好印象深刻的题,前天选拔赛给跪了。怪我这种关键题没敲出来。

题意很简单,给你一串无规则的数列,再给个m值,求出满足 数列和>=m的长度最小的连续子串。。。确实一开始卡住了,因为看数据肯定是nlogn的算法才能过,我和聪哥坐在外面的楼梯上,想了各种方法,我就一直在想前缀和以及 先求出最长连续和 再两边删减等等,刚提出来 就马上否定了。。后来还是聪哥想出个比较靠谱的方案,即二分结果,然后对二分出来的结果 先枚举右端点,再通过二分的结果得到左端点,找到左端点和右端点中间最小的前缀和,用左端点的前缀和 减去 最小的前缀和,如果结果>=m,意味着该二分结果可行,可以继续缩小下去,否则说明不行。

然后我就敲了,结果一直WA,后来知道自己哪里傻逼了,我对二分出来的len,在枚举右端点的时候,直接就是从len到n,0-(len-1)这里就没有枚举右端点,事实是,可能恰好结果就在0-(len-1),真是傻逼

不过这个算法确实是正确的,但还是有时限问题  因为多了个二分,其时间复杂度就变成了 n*logn*logn,所以只好聪哥改了我上面的漏洞之后,果然A了,不过用了1.7s,而且,之后我再提及他的同样的代码,就TLE了,估计是UVA的机器突然又跑慢了。所以这个算法还是有问题的。

后来跟聪哥讨论后觉得二分可能可以去掉不要,确实可以去掉。。。我又傻逼了,这么简单问题没想到,预处理完suffix前缀和之后,题目不就是要求suffix[右端点]-suffix[i]>=m,满足这个条件的 离右端点的最近的点,也就是说  我令 val=suffix[i]<=suffix[右端点]-m,即求离右端点最近的满足>=val的suffix所在的点,如果我线段树里面维护的是suffix的最小值,那么我为了求离右端点最近的,我肯定是对 线段树这样query:假如当前右子树满足 >=val,则优先走右子树,否则如何左子树>=val,则走左子树,如果左右都走不了 就是-1了

这样得到的值必定是离右端点最近的啊啊啊啊。

我真是傻逼

这是之前的思路 算法正确 但是会TLE

#include <iostream>
#include <cstdio>
#define N 500010
#define LL long long
#define lson rt<<1,l,mid
#define rson (rt<<1|1),mid+1,r
using namespace std;
LL suffix[N];
int n,m;
LL tree[N*3];
void build(int rt,int l,int r)
{
    if (l>=r){
        tree[rt]=suffix[l];
        return;
    }
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
    if (tree[rt<<1]>tree[rt<<1|1])
        tree[rt]=tree[rt<<1|1];
    else
        tree[rt]=tree[rt<<1];
}
LL query(int L,int R,int rt,int l,int r)
{
    if (L<=l && r<=R){
        return tree[rt];
    }
    int mid=(l+r)>>1;
    if (R<=mid){
     return query(L,R,lson);
    }
    if (L>mid){
     return query(L,R,rson);
    }
    return min(query(L,R,lson),query(L,R,rson));
}
int ok(int x)
{
    for (int i=2;i<=n;i++){ //这里应该从前面开始枚举,比赛的时候就是直接从len开始枚举,导致一直WA。。。虽然改了也不一定会避免时限超时,但总归这样是我敲错了,那样是我们思路错了,性质不一样。
        int j=i-x;
        if (j<0) j=0;
        LL mini;
        if (i==2) mini=0;
        else
         mini=query(j,i-2,1,0,n);
        if (suffix[i]-mini>=m) return 1;
    }
    return 0;
}

int main()
{
    int t;
    scanf("%d",&t);
    while (t--){
        suffix[0]=0;
        scanf("%d%d",&n,&m);
        bool isok=0;
        for (int i=1;i<=n;i++){
            scanf("%lld",&num[i]);
            suffix[i]=suffix[i-1]+num[i];
            if (num[i]>=m)  isok=1;
        }
        if (isok){
            puts("1");
            continue;
        }
        if (m<=0){
            puts("-1");
            continue;
        }
        build(1,0,n);
        int L=2,R=n+1,mid;
        bool flag=0;
        int ans=-1;
        while (L<R)
        {
            mid=(L+R)>>1;
            if (ok(mid)){
                flag=1;
                ans=mid;
                R=mid;
            }
            else
            L=mid+1;
        }
        if (!flag){
            puts("-1");
        }
        else
         printf("%d
",ans);
    }
    return 0;
}
/*
7 7
3 -2 1 6 1 -2 3
*/

AC代码

#include <iostream>
#include <cstdio>
#include <cstring>
#define N 500010
#define LL long long
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
using namespace std;
LL suffix[N];
int n,m;
LL tree[N*3];
void build(int rt,int l,int r)
{
    if (l>=r){
        tree[rt]=suffix[l];
        return;
    }
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
    if (tree[rt<<1]>tree[rt<<1|1])
        tree[rt]=tree[rt<<1|1];
    else
        tree[rt]=tree[rt<<1];
}
int query(LL val,int L,int R,int rt,int l,int r)
{
    if (l>=r)
        return l;
    int mid=(l+r)>>1;
    int ret1=-1,ret2=-1;
    if (mid<R) //因为是从顶点开始搜索起的,所以用L和R来控制区间
        if(val>=tree[rt<<1|1]){ //优先找右子树,就能得到最近的点
            ret1=query(val,L,R,rson);
        }
    if (L<=mid)
     if (val>=tree[rt<<1]){
         ret2=query(val,L,R,lson);
         }
    int ret=max(ret1,ret2);
    return ret;
}
int main()
{
    int t;
    scanf("%d",&t);
    while (t--){
        suffix[0]=0;
        scanf("%d%d",&n,&m);
        bool isok=0;
        for (int i=1;i<=n;i++){
            scanf("%lld",&suffix[i]);
            if (suffix[i]>=m)  isok=1;
            suffix[i]+=suffix[i-1];
        }
        if (isok){
            puts("1");
            continue;
        }
        if (m<=0){
            puts("-1");
            continue;
        }
        build(1,0,n);
        int ans=N;
        for (int i=2;i<=n;i++){
            LL v=suffix[i]-m;
            int res=query(v,0,i-1,1,0,n); //对每个v 在 0到 i-1的区间里,从线段树的顶点开始搜索起,得到离右端点最近的点。
            if (res!=-1) ans=min(ans,i-res);
           // cout<<i<<" "<<res<<endl;
        }
        if (ans==N) ans=-1;
        printf("%d
",ans);
    }
    return 0;
}
原文地址:https://www.cnblogs.com/kkrisen/p/3706695.html