您的位置:首页 > 其它

POJ 3264 Balanced Lineup-初入算法 线段树

2018-03-20 15:07 441 查看
题目:一个农场主有n头牛,每头牛都有一个高度,农场主给q次询问,希望求出询问给的区间中最大最小值的差
         条件:1,求区间最大最小值的差
输入:n(1<=n<=50000)     q(1<=q<=200000)
         n行牛的高度(1<=高度<=1000000)
         q次询问的区间
输出:每次询问后最大最小值的差
思路:题目已经提示了线段树,查阅了线段树的资料后,就是一道非常基础的线段树题目。只要懂得如何建立线段树和查询就可以了。先建立线段树,然后查询所要的区间的最大值和最小值。
代码:#include <stdio.h>
#include <stdlib.h>
#include <math.h>
#include <string.h>
#define max(a,b) a>b?a:b
#define min(a,b) a<b?a:b
int a[50005],segtree[50005*4][2];//一个存最小值,一个存最大值
int tree(int node,int begin,int end)//建立线段树
{
if(begin==end)//如果已经到了叶子的尽头,最大值和最小值就是本身
{
segtree[node][0]=a[begin];
segtree[node][1]=a[begin];
}
else
{
tree(2*node,begin,(begin+end)/2);//左儿子继续建立,直到尽头
tree(2*node+1,(begin+end)/2+1,end);//右儿子继续建立,直到尽头
if(segtree[2*node][0]<=segtree[2*node+1][0])//取左右儿子中的最小值
segtree[node][0]=segtree[2*node][0];
else
segtree[node][0]=segtree[2*node+1][0];
if(segtree[2*node][1]>segtree[2*node+1][1])//取左右儿子中的最大值
segtree[node][1]=segtree[2*node][1];
else
segtree[node][1]=segtree[2*node+1][1];
}
}
int query1(int node,int begin,int end,int left,int right)//访问最小值
{
int p1,p2;
if(left>end||right<begin)//不在左右儿子范围内
return -1;
if(begin>=left&&end<=right)//刚好在这个范围内
return segtree[node][0];
p1=query1(2*node,begin,(begin+end)/2,left,right);//继续访问
p2=query1(2*node+1,(begin+end)/2+1,end,left,right);
if(p1==-1)
return p2;
if(p2==-1)
return p1;
if(p1<=p2)//判断左右儿子那边的最小值更小
return p1;
return p2;
}
int query2(int node,int begin,int end,int left,int right)//查询最大值
{
int p1,p2;
if(left>end||right<begin)
return -1;
if(begin>=left&&end<=right)
return segtree[node][1];
p1=query2(2*node,begin,(begin+end)/2,left,right);
p2=query2(2*node+1,(begin+end)/2+1,end,left,right);
if(p1==-1)
return p2;
if(p2==-1)
return p1;
if(p1>=p2)
return p1;
return p2;
}
int main()
{
int n,m;
int i,j;
scanf("%d %d",&n,&m);
for(i=0;i<n;i++)
{
scanf("%d",&a[i]);
}
tree(1,0,n-1);
while(m--)
{
int k,h,sum;
scanf("%d %d",&k,&h);
sum=query2(1,0,n-1,k-1,h-1)-query1(1,0,n-1,k-1,h-1);
printf("%d\n",sum);
}
return 0;
}
总结:1,初步学习了线段树,只要沉下心来学,发现也没有想象中那么难懂
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: