1. 程式人生 > >POJ 3264 Balanced Lineup 線段樹的構建+區間查詢

POJ 3264 Balanced Lineup 線段樹的構建+區間查詢

第一次寫線段樹,能寫出來感覺很開心。。。

這道題屬於最簡單的線段樹問題,只有建樹與區間查詢兩大模組,後來我把單點更新的部分去掉了,建樹如果用單點更新的方法的話效率有點低。

首先說一下我的線段樹風格,我採用的是左閉右開區間,結點序號從1開始,陣列的0號元素空出來了,原因是這樣的話可以用位運算很方便地表示兩個兒子結點的序號,如果從0開始貌似得用加減運算了,有點慢。

//Run ID	User	Problem	Result	Memory	Time	Language	Code Length	Submit Time
//13076065	ritying	3264	Accepted	1164K	2297MS	C++	1850B	2014-07-14 15:44:05
#include <cstdio>
#include <iostream>
#include <cstring>
#include <climits>
using namespace std;
#define lson(x) ((x)<<1)
#define rson(x) ((x)<<1 | 1)

const int MAXN = 50000;
const int INF = INT_MAX;
int dat1[4*MAXN], dat2[4*MAXN]; //如果不幸運的話葉子節點將會是原來的兩倍,再加上中間節點所以一共是4倍(如果結點序號從1開始的話理論上是4*MAXN-4)
int n, N, Q;

void init(int N){
    n = 1;
    while(n < N) n <<= 1;
    for(int i=1; i<=n; i++){
        dat1[i] = INF;
        dat2[i] = 0;
    }
}

inline void PushUp(int rt){
    dat1[rt] = min(dat1[lson(rt)], dat1[rson(rt)]);
    dat2[rt] = max(dat2[lson(rt)], dat2[rson(rt)]);
}

void build(int rt, int l, int r){
    if(l+1 == r){
        if(l <= N){
           scanf("%d", &dat1[rt]);
           dat2[rt] = dat1[rt];
        }
        else{
            dat1[rt] = INF;
            dat2[rt] = 0;
        }
        return;
    }
    build(lson(rt), l, (l+r)>>1);
    build(rson(rt), (l+r)>>1, r);
    PushUp(rt);
}

int querymin(int a, int b, int k, int l, int r){
    if(b <= l || r <= a) return INF;
    else if(a <= l && r <= b) return dat1[k];
    else{
        int vl = querymin(a, b, lson(k), l, (l+r)>>1);
        int vr = querymin(a, b, rson(k), (l+r)>>1, r);
        return min(vl, vr);
    }
}

int querymax(int a, int b, int k, int l, int r){
    if(b <= l || r <= a) return 0;
    else if(a <= l && r <= b) return dat2[k];
    else{
        int vl = querymax(a, b, lson(k), l, (l+r)>>1);
        int vr = querymax(a, b, rson(k), (l+r)>>1, r);
        return max(vl, vr);
    }
}

int main(){
    while(scanf("%d %d", &N, &Q) != EOF){
       init(N);
       build(1, 1, n+1);
       while(Q--){
           int a, b;
           scanf("%d %d", &a, &b);
           int low = querymin(a, b+1, 1, 1, n+1);
           int high = querymax(a, b+1, 1, 1, n+1);
           printf("%d\n", high - low);
       }
    }
    return 0;
}