0

我正在尝试构建用于执行 RMQ 的段树。不知何故,无论我查询什么范围,它都会返回 0。

例如,我的数组是[ 1,2,3,4,5,6,7,8,9,10 ]. 从索引 3 到 5 的 RMQ 应该给出 4。但我的代码一直输出 0。

我的代码:

#include<bits/stdc++.h>

using namespace std;

#define ll long long

ll N, st[2*100000], arr[100000];

void build(int r, int lo,int hi) {
    if(lo==hi) {
        st[r] = arr[lo];
    } else {
        build(r<<1,lo,(lo+hi)>>1);
        build((r<<1)|1,((lo+hi)>>1)+1,hi);
        st[r] = min(st[r<<1],st[(r<<1)|1]);
    }
}

void update(int r,int lo,int hi,ll val,int i) {
    if(lo==hi) {
        st[r] = val;
    } else {
        int mid = (lo+hi)>>1;
        if(lo <= i && i <= mid) {
            update(r<<1,lo,mid,val,i);
        } else {
            update((r<<1)|1,mid+1,hi,val,i);
        }
        st[r] = min(st[r<<1],st[(r<<1)|1]);
    }   
}

ll query(int r,int lo,int hi,int a,int b) {
    if(lo > b || hi < a) {
        return LLONG_MAX;
    } else if(lo <= a && hi >= b) {
        return st[r];
    } else {
        int mid = (lo+hi)>>1;
        return st[r] = min(query(r<<1,lo,mid,a,mid),query((r<<1)|1,mid+1,hi,mid+1,b));
    }
}

int main(void) {
    for(int i = 0; i <= 10; i++) arr[i] = i;
    build(1,0,10);
    cout << query(1,0,10,3,5) << endl;
    return 0;
}

编辑:

感谢所有的帮助。就像下面提到的答案一样,我的查询范围是错误的。除此之外还有2个错误:

if(lo <= a && hi >= b)应该(a <= lo && b >= hi)

return st[r] = min(query(r<<1,lo,mid,a,mid),query((r<<1)|1,mid+1,hi,mid+1,b));

应该

return min(query(r<<1,lo,mid,a,mid),query((r<<1)|1,mid+1,hi,mid+1,b));

因为我只是进行查询而不应该修改树。

4

1 回答 1

2

这行代码:

return st[r] = min(query(r<<1,lo,mid,a,mid),query((r<<1)|1,mid+1,hi,mid+1,b));

你为什么要改变你的ab简而言之,你在每次递归调用中都改变你的查询。它应该是

 return st[r] = min(query(r<<1,lo,mid,a,b),query((r<<1)|1,mid+1,hi,a,b));

而且你的更新功能是错误的:

在这里检查

于 2016-12-05T17:33:49.630 回答