POJ3264 Balanced Lineup 线段树||RMQ

来源:互联网 发布:云计算与电子政务 编辑:程序博客网 时间:2024/06/07 19:40

题目链接:http://poj.org/problem?id=3264


题目大意:给出一个序列,Q次查询,每次查询找出该区间内最大值和最小值的差。


分析:线段树和RMQ都可以。


线段树实现代码如下(2412K,2188MS):

#include <cstdio>#include <cstring>#include <iostream>using namespace std;const int maxn=50005;struct segment{    int l,r;    int maxi,mini;};segment tree[maxn<<2];int height[maxn];int max(int a,int b){    return a>b?a:b;}int min(int a,int b){    return a<b?a:b;}void build(int root,int l,int r)//初始化线段树{    tree[root].l=l;    tree[root].r=r;    if(tree[root].l==tree[root].r)    {        tree[root].maxi=height[l];        tree[root].mini=height[l];        return ;    }    int mid=(l+r)>>1;    build(root<<1,l,mid);    build(root<<1|1,mid+1,r);    tree[root].maxi=max(tree[root<<1].maxi,tree[root<<1|1].maxi);    tree[root].mini=min(tree[root<<1].mini,tree[root<<1|1].mini);}int query_max(int root,int l,int r)//查询区间最大值{    if(r<tree[root].l||l>tree[root].r)      return 0;    if(l<=tree[root].l&&r>=tree[root].r)      return tree[root].maxi;    int a=query_max(root<<1,l,r);    int b=query_max(root<<1|1,l,r);    return max(a,b);}int query_min(int root,int l,int r)//查询区间最小值{    if(r<tree[root].l||l>tree[root].r)      return 99999999;//不在改区间则返回正无穷    if(l<=tree[root].l&&r>=tree[root].r)      return tree[root].mini;    int a=query_min(root<<1,l,r);    int b=query_min(root<<1|1,l,r);    return min(a,b);}int main(){    int n,q;    while(scanf("%d%d",&n,&q)!=-1)    {        for(int i=1;i<=n;i++)          scanf("%d",&height[i]);        build(1,1,n);        while(q--)        {            int a,b;            scanf("%d%d",&a,&b);            int ans=query_max(1,a,b)-query_min(1,a,b);            printf("%d\n",ans);        }    }    return 0;}



RMQ实现代码如下(8192K,1610MS):

#include <cstdio>#include <iostream>using namespace std;const int maxn=50005;int height[maxn];int fmax[maxn][20];int fmin[maxn][20];int max(int a,int b){    return a>b?a:b;}int min(int a,int b){    return a<b?a:b;}void init_max(int n){    for(int i=1;i<=n;i++) fmax[i][0]=height[i];    for(int k=1;(1<<k)<=n;k++)      for(int i=1;i+(1<<k)-1<=n;i++)        fmax[i][k]=max(fmax[i][k-1],fmax[i+(1<<(k-1))][k-1]);}void init_min(int n){    for(int i=1;i<=n;i++) fmin[i][0]=height[i];    for(int k=1;(1<<k)<=n;k++)      for(int i=1;i+(1<<k)-1<=n;i++)        fmin[i][k]=min(fmin[i][k-1],fmin[i+(1<<(k-1))][k-1]);}int query_max(int l,int r){    int k=0;    while((1<<(k+1))<=r-l+1) k++;    return max(fmax[l][k],fmax[r-(1<<k)+1][k]);}int query_min(int l,int r){    int k=0;    while((1<<(k+1))<=r-l+1) k++;    return min(fmin[l][k],fmin[r-(1<<k)+1][k]);}int main(){    int n,q;    while(scanf("%d%d",&n,&q)!=-1)    {        for(int i=1;i<=n;i++)          scanf("%d",&height[i]);        init_max(n);        init_min(n);        while(q--)        {            int a,b;            scanf("%d%d",&a,&b);            printf("%d\n",query_max(a,b)-query_min(a,b));        }    }    return 0;}


0 0