欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

线段树入门-hdu1754

程序员文章站 2024-03-20 09:45:28
...
题目:
很多学校流行一种比较的习惯。老师们很喜欢询问,从某某到某某当中,分数最高的是多少。 
这让很多学生很反感。 

不管你喜不喜欢,现在需要你做的是,就是按照老师的要求,写一个程序,模拟老师的询问。当然,老师有时候需要更新某位同学的成绩。
Input本题目包含多组测试,请处理到文件结束。 
在每个测试的第一行,有两个正整数 N 和 M ( 0<N<=200000,0<M<5000 ),分别代表学生的数目和操作的数目。 
学生ID编号分别从1编到N。 
第二行包含N个整数,代表这N个学生的初始成绩,其中第i个数代表ID为i的学生的成绩。 
接下来有M行。每一行有一个字符 C (只取'Q'或'U') ,和两个正整数A,B。 
当C为'Q'的时候,表示这是一条询问操作,它询问ID从A到B(包括A,B)的学生当中,成绩最高的是多少。 
当C为'U'的时候,表示这是一条更新操作,要求把ID为A的学生的成绩更改为B。 
Output对于每一次询问操作,在一行里面输出最高成绩。Sample Input
5 6
1 2 3 4 5
Q 1 5
U 3 6
Q 3 4
Q 4 5
U 2 9
Q 1 5
Sample Output
5
6
5
9

线段树:

参考:https://blog.csdn.net/zearot/article/details/52280189

这里和参考里不一样的是,这里是求一段区间的最小值,所以在Query的时候,应该返回左右子树的最大值。同时PushUp()函数也不太一样。这里注意scanf的输入问题,当输入整型后突然输入字符,需要getchar()一下,把缓存区清理。

#include <iostream>
#include<stdio.h>
using namespace std;
#define maxn 200008
int A[maxn];
int sum[maxn*4];
void PushUp(int rt)
{  //这里是求最大值,rt表示当前节点的实际存储位置,下标
    sum[rt]=max(sum[rt<<1],sum[rt<<1|1]);
}
void build(int l,int r,int rt)
{
    //自顶向下
    if(l==r)
    {sum[rt]=A[l];
   // cout<<"rt:"<<rt<<" "<<sum[rt]<<" ";
    return;
    }
    int mid=(l+r)/2;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    PushUp(rt);//左右子树建好后,更新本节点信息
}
int Query(int L,int R,int l ,int r,int rt)
{
    //L,R是查询区间,不会改变
    if(L<=l&&R>=r)
        return sum[rt];
   int  mid=(l+r)/2;
 int ANS1=0,ANS2=0;
    //因为一开始[L,R]肯定是在[l,r]里边,
    //所以不会出现[L,R]跑到[l,r]外边的情况
    if(L<=mid)  //和左子树区间有交集
        ANS1=Query(L,R,l,mid,rt<<1);
    if(R>mid)
        ANS2=Query(L,R,mid+1,r,rt<<1|1);
    return max(ANS1,ANS2);

}
void Update(int L,int C,int l,int r,int rt)
{

    if(l==r)
       {
        sum[rt]=C;
        return;
       }
    int mid=(l+r)/2;
    if(L<=mid)
        Update(L,C,l,mid,rt<<1);
    else
        Update(L,C,mid+1,r,rt<<1|1);
    PushUp(rt);//更新本节点
}
int main()
{
   // int r=1<<2+1,t=1<<2|1;//+优先级高
   // cout<<r<<t<<endl;
   int N,M;
   while(~scanf("%d%d",&N,&M))
  {


   for(int i=1;i<=N;i++)
    scanf("%d",&A[i]);
   build(1,N,1);

   char C;
   int a,b;
   while(M--)
   {
        getchar();//注意字符和整型切换
       scanf("%c%d%d",&C,&a,&b);
       //printf("%c %d %d\n",C,a,b);
       if(C=='Q')
       {
           int ans=Query(a,b,1,N,1);
           printf("%d\n",ans);

       }
       if(C=='U')
       {
           Update(a,b,1,N,1);
         //  for(int j=1;j<=9;j++)
           // cout<<sum[j]<<" ";
           //cout<<endl;
       }

   }
  }
    return 0;
}