poj 3264 Balanced Lineup (线段树)

来源:互联网 发布:软件著作权 代码格式 编辑:程序博客网 时间:2024/06/05 00:15

题意:http://poj.org/problem?id=3264

这里需要用到线段树的查询,不过特别的是,它需要查找两个值,最大值和最小值,在查找操作中稍加处理即可。

#include <iostream>#include<cstdio>#include<algorithm>#include<cstring>using namespace std;const int maxn=5e4+5;int height[maxn];// 最长区间长度struct node{    int left,right,high,low;}btree[3*maxn]; // 设置根节点下标是1,则左孩子是2*dex,右孩子下标是2*dex+1.void build(int root,int l,int r){  //主函数数组里的dex对应这里的l,r    btree[root].left=l;  //root对应树和各种子树的根节点    btree[root].right=r;    if(l==r){        btree[root].low=btree[root].high=height[l];        return ;    }    int mid=(l+r)/2;    build(root*2,l,mid);    build(root*2+1,mid+1,r);    btree[root].high=max(btree[2*root].high,btree[2*root+1].high); //递归回溯赋值给双亲结点    btree[root].low=min(btree[2*root].low,btree[2*root+1].low);}void query(int root,int L,int R,int &higher,int &lower){    if(L<=btree[root].left&&R>=btree[root].right){        higher=max(btree[root].high,higher);        lower=min(btree[root].low,lower);        return ;    }    int mid=(btree[root].left+btree[root].right)/2;    if(L<=mid)query(root*2,L,R,higher,lower); // L,R不要弄错,否则会数组越界    if(R>mid)query(2*root+1,L,R,higher,lower);}int main(int argc, char *argv[]) {    //freopen("cin.txt","r",stdin);    int N,Q,i;    while(cin>>N>>Q){        for(i=1;i<=N;i++)scanf("%d",&height[i]);        build(1,1,N);        for(i=1;i<=Q;i++){            int a,b,higher=0,lower=(int)(1e6+1);            scanf("%d%d",&a,&b);            query(1,a,b,higher,lower);            printf("%d\n",higher-lower);        }    }    return 0;}


0 0