HDUoj 1754 I Hate It ( 线段树

I Hate It

Time Limit: 9000/3000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 72934    Accepted Submission(s): 28152

Problem Description

很多学校流行一种比较的习惯。老师们很喜欢询问，从某某到某某当中，分数最高的是多少。
这让很多学生很反感。

不管你喜不喜欢，现在需要你做的是，就是按照老师的要求，写一个程序，模拟老师的询问。当然，老师有时候需要更新某位同学的成绩。

 

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
  
  

线段树模板题

#include <bits/stdc++.h>
using namespace std;
const int N = 200000*4+10;
struct node 
{
    int left, right;  
    int sum;   
}tree[N];
int arr[N];
void build(int l,int r,int step)
{
    tree[step].left = l;   
    tree[step].right = r;
    tree[step].sum = 0;
    if(l == r) {
        tree[step].sum = arr[l];
        return ;    
    }
    int mid = (l+r) >> 1; 
    build(l,mid,step<<1); 
    build(mid+1,r,step<<1|1);
    tree[step].sum = max(tree[step<<1].sum,tree[step<<1|1].sum);
} 
void update(int l, int r, int step)
{
    if(tree[step].left==l && tree[step].right==l) {
        tree[step].sum = r;
        return ;
    }

    int mid = (tree[step].left+tree[step].right) >> 1;

    if(l <= mid) update(l,r,step<<1);
    if(l > mid)  update(l,r,step<<1|1);
    tree[step].sum = max(tree[step<<1].sum,tree[step<<1|1].sum);
} 
int query(int l, int r,int step)
{
    //找到叶子返回值
    if(l==tree[step].left && r==tree[step].right) return tree[step].sum;
    int mid = (tree[step].left+tree[step].right) >> 1;
    //和更新类似
    if(r <= mid) return query(l,r,step<<1);
    if(l > mid) return query(l,r,step<<1|1);
    else
        return max(query(l,mid,step<<1),query(mid+1,r,step<<1|1));

}

int main()
{
    int n, m;
    char ch;
    while(~scanf("%d%d",&n,&m)) {
        for(int i = 1;i <= n; i++) {
            scanf("%d",&arr[i]); 
        }
        build(1,n,1);
        int u, v;
        while(m--) {
            getchar();
            scanf("%c%d%d",&ch,&u,&v);
            if(ch == 'Q') 
                printf("%d\n",query(u,v,1));
            if(ch == 'U')
                update(u,v,1);
        }
    }
return 0;
}