码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2015-05-03 12:04:21      阅读:141      评论:0      收藏:0      [点我收藏+]

标签:acm   poj   

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


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

标签:acm   poj   

原文地址:http://blog.csdn.net/weizhuwyzc000/article/details/45457313

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!