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

来源：互联网发布：养什么蛇不犯法知乎编辑：程序博客网时间：2024/06/11 01:04

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

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

#include<cstdio>#include<cstring>#include<iostream>#include<algorithm>#include<queue>#include<vector>#include<map>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(n<n_) n *= 2;    for(int i=0;i<2*n-1;i++) dat[i] = INF;}void update(int k,int a){    k += n - 1;    dat[k] = a;    while(k>0) {        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;}

2 0