POJ3264 Balanced Lineup 线段树||RMQ

题目链接: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;
}


你可能感兴趣的:(POJ3264 Balanced Lineup 线段树||RMQ)