F - I Hate It
很多学校流行一种比较的习惯。老师们很喜欢询问,从某某到某某当中,分数最高的是多少。
这让很多学生很反感。
不管你喜不喜欢,现在需要你做的是,就是按照老师的要求,写一个程序,模拟老师的询问。当然,老师有时候需要更新某位同学的成绩。
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
Hint
Huge input,the C function scanf() will work better than cin
思路,线段树的入门
code:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN =200000 + 10;
const int MINF = 0x80000000;
struct node{
int ll,rr;
int max;
}tree[MAXN*4];
int arr[MAXN];
//作弊又开
void create(int root,int left,int right){
tree[root].ll = left;
tree[root].rr = right;
//叶节点
if(right - left == 1) {
tree[root].max = arr[left]; return;
}
int mid = (left+right)/2;
create(root*2,left,mid);
create(root*2+1,mid,right);
//更新父节点
tree[root].max = max(tree[root*2+1].max,tree[root*2].max);
}
//从叶节点到父节点从新比较
//当当前遍历的区间包含更新区间时,进行更新
void update2(int root,int left,int right){
if(left>=tree[root].rr || right<=tree[root].ll ) return;
if(tree[root].rr - tree[root].ll == 1) {
tree[root].max = arr[left]; return;
}
update2(root*2,left,right);
update2(root*2+1,left,right);
tree[root].max = max(tree[root*2+1].max,tree[root*2].max);
}
//查询区间
int query(int root,int left,int right){
if(left>=tree[root].rr || right<=tree[root].ll ) return MINF;
if(left<=tree[root].ll && right>=tree[root].rr ) {
return tree[root].max;
}
int ansa = query(root*2,left,right);
int ansb = query(root*2+1,left,right);
return max(ansa,ansb);
}
int N,M;
int main(){
while(~scanf("%d%d",&N,&M)){
for(int i=1;i<=N;i++){
scanf("%d",&arr[i]);
}
memset(tree,0,sizeof(tree));
create(1,1,N+1);
for(int i=1;i<=M;i++){
char task;
int a,b;
getchar();
scanf("%c %d %d",&task,&a,&b);
// printf("--%c %d %d\n",task,a,b);
if(task == 'Q'){
//查询
int ans = query(1,a,b+1);
printf("%d\n",ans);
}else{
//更新
arr[a] = b;
update2(1,a,a+1);
}
}
}
return 0;
}
总结:
注意查询区间和更新之间的差别,这里更新是对子区间重新更新,如果遍历的当前节点的区间包含更新区间,那么就需要更新