【hdu1754】I Hate It——线段树模板题

I Hate It

Time Limit: 9000/3000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)

Total Submission(s): 85795    Accepted Submission(s): 32921

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


   
   

    
    

     
     Hint
    
    
Huge input,the C function scanf() will work better than cin
   
   
   
    
  
  

原文链接：http://acm.hdu.edu.cn/showproblem.php?pid=1754

【题意】：看不懂中文的请出门右转

【分析】：线段树的单点更新、区间最值查询

ac代码如下：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
using namespace std;

#define ls (root<<1)
#define rs (root<<1|1)

using namespace std;
const int N = 200000;

struct node
{
    long long l,r;
    long long maxn;
    long long add;//lazy标记
    long long mid()
    {
        return (l+r)>>1;
    }
}seg[N<<2];
long long num[N];

void pushup(int root)
{
    seg[root].maxn = max(seg[ls].maxn,seg[rs].maxn);//从左右子树向上返回最大值
}

void build(int l,int r,int root)
{
    seg[root].l=l;
    seg[root].r=r;
    seg[root].add=0;
    if(l == r)//递归到叶子节点无法继续递归，所以返回
    {
        seg[root].maxn = num[l];
        return ;
    }
    long long mid = seg[root].mid();
    build(l,mid,ls);//不断递归建立左子树
    build(mid+1,r,rs);//同理
    pushup(root);
}

void pushdown(int root)//lazy标记下传
{
    if(seg[root].add)
    {
        if(seg[root].l==seg[root].r)
        {
            seg[root].add=0;
            return ;
        }
        seg[ls].add  = seg[root].add;
        seg[rs].add  = seg[root].add;
    }
}

void update(int l,int change,int root)
{
    if(seg[root].l == l&&seg[root].r == l)//找到树中对应的节点
    {
        seg[root].add  = change;
        seg[root].maxn = change;
        return;
    }
    if(seg[root].l == seg[root].r)//如果是叶子节点，最大值就是它本身
    {
        seg[root].maxn = change;
        return;
    }
    pushdown(root);
    int mid = seg[root].mid();
    if(l<=mid)      update(l,change,ls);//更新左子树
    else if(l>mid)  update(l,change,rs);//更新右子树
    pushup(root);
}

long long query_max(int l,int r,int root)
{
    if(seg[root].l >= l&&seg[root].r <= r)//树的区间在查询区间内
    {
        return seg[root].maxn;
    }
    pushdown(root);
    long long ans;
    long long mid = seg[root].mid();
    if(r<=mid)      ans = query_max(l,r,ls);//查询区间在左子树则向左查询
    else if(l>mid)  ans = query_max(l,r,rs);//同理
    else
    {
        ans=max(query_max(l,mid,ls),query_max(mid+1,r,rs));//从叶子节点返回答案
    }
    return ans;
}

int main()
{
    int n, m, l, r;
    while(~scanf("%d%d",&n,&m))
    {
        for(int i = 1;i <= n; i++)
        {
            scanf("%d",&num[i]);
        }
        build(1,n,1);
        for(int i = 1;i <= m; i++)
        {
            char ch;
            scanf(" %c",&ch);
            scanf("%d%d",&l,&r);
            if(ch == 'Q')
            {
                cout << query_max(l,r,1) <<endl;
            }
            else if(ch == 'U')
            {
                update(l,r,1);
            }
        }
    }
    return 0;
}