Balanced Lineup(POJ-3264)(线段树)

很基础的一道线段树的题,有个地方卡了我好久,我下面的这个代码所求的区间是左闭右开的,所以如果所求区间包括区间端点的话需要在右区间上+1

线段树是一种高效的数据结构,特点是求一个区间里的最小、最大值。      数据结构感觉像模板,但是其中的思想应该更值得我们学习,不过话说现在也没几个人能静下心去研究它的原理了吧。。

#include
#include
#include
#include
#include
#include
#include
using namespace std;
const int INF = 1000000+10;
const int max_n = 2*50000+100;
int n,m,a,b,q,dat[max_n*2-1],dat2[max_n*2-1];
void init(int n_) {
    n = 1;
    while(n0) {
        k=(k-1)/2;
        dat[k] = min(dat[k*2+1],dat[k*2+2]);
    }
}
void init2(int n_){
    for(int i=0;i<2*n-1;i++) dat2[i] = -100;
}
void update2(int k,int a){
    k += n-1;
    dat2[k] = a;
    while(k>0){
        k = (k-1)/2;
        dat2[k] = max(dat2[k*2+1],dat2[k*2+2]);
    }
}
int query(int a,int b,int k,int l,int r) {
    if(r<=a||b<=l) return INF;

    if(a<=l&&r<=b) return dat[k];
    else {
        int vl = query(a,b,k*2+1,l,(l+r)/2);
        int vr = query(a,b,k*2+2,(l+r)/2,r);
        return min(vl,vr);
    }
}
int query2(int a,int b,int k,int l,int r){
    if(r<=a||b<=l) return -1;
    if(a<=l&&r<=b) return dat2[k];
    else {
        int vl = query2(a,b,k*2+1,l,(l+r)/2);
        int vr = query2(a,b,k*2+2,(l+r)/2,r);
        return max(vl,vr);
    }
}
int main() {
    scanf("%d%d",&m,&q);
    init(m);
    init2(m);
    for(int i=1;i<=m;i++) {
        scanf("%d",&a);
        update(i,a);
        update2(i,a);
    }
    int maxn = 1000000000;
    while(q--) {
        scanf("%d%d",&a,&b);
        int v = query(a,b+1,0,0,n);
        int u = query2(a,b+1,0,0,n);
        printf("%d\n",u-v);
    }
    return 0;
}


你可能感兴趣的:(线段树和树状数组)