HDU 1754 I Hate It【线段树】【单点更新】

HDU 1754 I Hate It

#include <bits/stdc++.h>
using namespace std;

#define ls o<<1   //左子节点
#define rs o<<1|1  //右子节点
//二进制位左移一位代表着数值 *2 ，而如果左移完之后再或上 1 ，由于左移完之后最后一位二进制位上一定会是 0 ，所以 |1等价于 +1 。 

const int MAXN = 2e5+10;

int a[MAXN];

struct node 
{
    int l,r;//  一个节点 所代表的 区间左值和区间右值
    int max;
}t[MAXN<<2];//节点数四倍 

inline void push_up(int o) //合并两个子节点的信息 
{
    t[o].max = max(t[ls].max,t[rs].max);//父节点的值为两子节点的最大值 
}
void build(int l,int r,int o) //建树 .o是节点编号 
{
    t[o].l=l,//获取当前节点的区间左值 
    t[o].r=r;//获取 当前节点的区间右值
    if(l==r) // 区间左值==区间右值，说明到达叶子节点
        {
            t[o].max=a[l]; 
            return;
        }
    int mid=(l+r)>>1;//二分存中点
    build(l,mid,ls); //建左子树 
    build(mid+1,r,rs);//建右子树 
    push_up(o);//此处由于我们是要通过子节点来维护父亲节点，所以push_up的位置应当是在回溯时。 
}
inline void update(int s, int val, int o) //更新 .s为需要修改的叶子结点左端点,val是 这个营地的人数 ,o为当前更新到的结点
{
    if(t[o].l==t[o].r && t[o].l==s) //??? t[o].l==s
    {
        t[o].max=val;
        return;
    }
    int mid=(t[o].l+t[o].r)>>1;
    if(s<=mid) //若需要更新的叶子结点在当前结点的左儿子结点的范围内，则递归更新左儿子结点
        update(s,val,ls);                    //更新左子树
    else 
        update(s,val,rs);                    //更新右子树
    push_up(o);
}
inline int query(int l,int r, int o) //查询 
{
    if(t[o].l>=l&&t[o].r<=r) // 完全包含才可以判出
        return t[o].max;
    int mid=(t[o].l+t[o].r)>>1;
     //   由于要查询的区间不同 ， 可能导致的 三种不同的 方向 
    if(r<=mid)                         //查询左子树
        return query(l,r,ls);
    else if(l>mid)                     //查询右子树
        return query(l,r,rs);
    else                             //查询区间横跨左右子树，两棵子树一起查询
    {
        return max(query(l,mid,ls),query(mid+1,r,rs));
    }
}
int main() 
{
    int n,m,l,r; 
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        
        for(int i = 1;i<=n;++i) 
            scanf("%d",&a[i]);
        build(1,n,1);
        char ch[10];
        while(m--)
        {
            scanf("%s%d%d",ch,&l,&r);
            if(ch[0]=='U')
                update(l,r,1);
            else 
                cout<<query(l,r,1)<<endl;
        }   
    }
return 0;
}