B - I Hate It HDU - 1754( 线段树变例 )
很多学校流行一种比较的习惯。老师们很喜欢询问,从某某到某某当中,分数最高的是多少。
这让很多学生很反感。
不管你喜不喜欢,现在需要你做的是,就是按照老师的要求,写一个程序,模拟老师的询问。当然,老师有时候需要更新某位同学的成绩。
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
思路:把线段树的累加条件换成求最大值的条件。
代码:
#include <iostream>
#include <algorithm>
#include <math.h>
#include <string.h>
#include <stdio.h>
using namespace std;
int a[200005];
int tree[800005];
void built_tree( int node, int left, int right )
{
if ( left==right ) {
tree[node] = a[left];
return ;
}
int mid = (left+right)/2;
int node_left = 2*node + 1;
int node_right = 2*node + 2;
built_tree(node_left,left,mid);
built_tree(node_right,mid+1,right);
tree[node] = max(tree[node_left],tree[node_right]);
}
void update( int node, int left, int right, int idx, int val )
{
if ( left==right ) {
a[idx] = val;
tree[node] = val;
return ;
}
int mid = (left+right)/2;
int node_left = 2*node + 1;
int node_right = 2*node + 2;
if ( idx>mid ) {
update(node_right,mid+1,right,idx,val);
}
else {
update(node_left,left,mid,idx,val);
}
tree[node] = max(tree[node_left],tree[node_right]);
}
int query( int node, int left, int right,int L,int R )
{
if ( left>R || right<L ) {
return 0;
}
else if ( left>=L && right<=R ) {
return tree[node];
}
else if ( left==right ) {
return tree[node];
}
int mid = (left+right)/2;
int node_left = 2*node + 1;
int node_right = 2*node + 2;
int sum_left = query(node_left,left,mid,L,R);
int sum_right = query(node_right,mid+1,right,L,R);
return max(sum_left,sum_right);
}
int main()
{
int n,m,i;
char str;
int a1,b1;
while ( ~scanf("%d %d",&n,&m) ) {
for ( i=0; i<n; i++ ) {
scanf("%d",&a[i]);
}
built_tree(0,0,n-1);
while ( m-- ) {
cin >> str;
scanf("%d %d",&a1,&b1);
if ( str=='Q' ) {
int ans = query(0,0,n-1,a1-1,b1-1);
printf("%d\n",ans);
}
else if ( str=='U' ) {
update(0,0,n-1,a1-1,b1);
}
}
}
return 0;
}