线段树入门-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 5Sample 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;
}