POJ 3264 Balanced Lineup 線段樹的構建+區間查詢
阿新 • • 發佈:2019-01-23
第一次寫線段樹,能寫出來感覺很開心。。。
這道題屬於最簡單的線段樹問題,只有建樹與區間查詢兩大模組,後來我把單點更新的部分去掉了,建樹如果用單點更新的方法的話效率有點低。
首先說一下我的線段樹風格,我採用的是左閉右開區間,結點序號從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; }