hdu 1754 I Hate It

线段树入门题

中文题就不说题意了。就是基本的区间询问最大值和修改某个特定单元的值，初学线段树，自己硬生生啃出来的代码，写得不好，不过1Y了这道题很是鼓舞。时间上500ms，不知道那些100ms是怎么做到的

 

#include <cstdio>
#include <cstring>
#define N 200010
#define INF 0x3f3f3f3f
struct segment_tree
{
    int a,b;  //区间
    int l,r;  //左右孩子
    int max;
}tree[2*N];
int a[N];
int tot; //线段树结点的个数
int n,m;  //区间长度和询问的个数

int query(int num , int a , int b) //区间[a,b]
{
    //if(tree[num].b<a || tree[num].a>b)  //该结点的区间和需要询问的区间不想交
        //return -INF;  

    if(tree[num].a==a && tree[num].b==b)  //完全重合直接返回最大值
        return tree[num].max;

    int mid,max1,max2;

    mid=(tree[num].a+tree[num].b)/2;
    max1=max2=-INF;
    
    if(a>mid)  //不需要在左半区间中查找
        max2=query(tree[num].r,a,b);

    else if(b<=mid)  //不需要在右半区间中查找
        max1=query(tree[num].l,a,b);

    else      //横跨两个部分
    {
        max1=query(tree[num].l,a,mid);
        max2=query(tree[num].r,mid+1,b);
    }

    return max1>max2?max1:max2;
}

void updata(int num , int p ,int e)  //要将p位置的值变为e
{
    if(tree[num].a==tree[num].b)  
    {
        tree[num].max=e;
        return ;
    }

    int mid=(tree[num].a+tree[num].b)/2 , max,tmp;
    if(p<=mid)  //去左半区间查找
    {
        updata(tree[num].l , p ,e);
        tmp=tree[num].l;
    }
    else        //去右半区间查找
    {
        updata(tree[num].r , p , e);
        tmp=tree[num].r;
    }

    max=tree[tmp].max ;
    if(max>tree[num].max) tree[num].max=max;
    return ;
}
int create_tree(int x ,int y , int num)
{
    if(x==y)  //元结点
    {
        tree[num].a=tree[num].b=x;
        tree[num].l=tree[num].r=-1;
        return tree[num].max=a[x];
    }

    int mid=(x+y)/2,max1,max2,num1,num2;

    tree[num].a=x;
    tree[num].b=y;
    num1=tot+1; num2=tot+2;
    tree[num].l=++tot;
    tree[num].r=++tot;
    max1=create_tree(x,mid,num1);
    max2=create_tree(mid+1,y,num2);
    return tree[num].max=max1>max2?max1:max2;
}

int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(int i=1; i<=n; i++) scanf("%d",&a[i]);
        tot=1;
        create_tree(1,n,1);  //建树
        while(m--)  //读入m个询问
        {
            char s[5]; int tmp1,tmp2;
            scanf("%s%d%d",s,&tmp1,&tmp2);
            if(!strcmp(s,"Q"))  //询问区间的最大值
            {
                int ans=query(1,tmp1,tmp2); //区间为[tmp1,tmp2],从第1个结点开始
                printf("%d\n",ans);
            }
            else               //更改数值
                updata(1,tmp1,tmp2);
        }
    }
    return 0;
}