HDOJ 1754 I Hate It(线段树)

 I Hate It

Time Limit:3000MS     Memory Limit:32768KB     64bit IO Format:%I64d & %I64u

Submit  Status

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

对于每一次询问操作，在一行里面输出最高成绩。

一看需要维护一个最大值和和值，需要更新单节点，线段树裸题，这里来学一下线段树，线段树是一种二叉搜索树，完全二叉树，以父区间的mid为分界点，划为左区间右区间，一直划分到叶子节点，一共有n个叶子节点。先来看建树，采用递归的方式来建树，以mid来划分区间，以id*2为左儿子，id*2+1为右儿子。

#define maxn 200005
int a[maxn];
struct node
{
    int id;
    int l;
    int r;
    int maxans;   //这里我们维护的是最大值
}tree[4*maxn];
void build(int id,int l,int r)   //建树
{
    tree[id].l=l;           //给每个节点的区间赋值，确定节点的负责的区间范围
    tree[id].r=r;
    if(l==r)                 //叶子节点赋值
    {
        tree[id].maxans=a[l];
        return;
    }
    int mid=(l+r)/2;           //左右儿子
    build(id*2,l,mid);
    build(id*2+1,mid+1,r);
    tree[id].maxans=max(tree[id*2].maxans,tree[id*2+1].maxans);    //当一个子树的左右儿子的最值都确定过后，父节点的最值为儿子的最大值，更新父节点；
}

然后再来看维护区间，这里用的是单点更新，每次更新一个只需要节点，我们就一直递归到要找的那个节点，然后更新，在回溯的时候要注意更新父节点;

void update(int id,int x,int newval)
{
    int l=tree[id].l;
    int r=tree[id].r;
    int mid=(l+r)/2;
    if(l==r)                  //找到这个节点l=r=x
    {
        tree[id].maxans=newval;
        return;
    }
    if(x<=mid)                    //划分区间
        update(id*2,x,newval);
    else
        update(id*2+1,x,newval);
    tree[id].maxans=max(tree[id*2].maxans,tree[id*2+1].maxans);       //儿子更新完后要更新父节点;
}

最后来看查找，这里是查找某一个区间中的最大值，在线段树中每个节点都有一个负责的区间范围，我们将要查找的区间和该节点的负责区间进行比较，如果这个节点的区间完全在所要查找的区间里面，那么这个节点的最大值就是我们需要的信息，返回该信息，如果要查找的区间在节点的左区间里面，那么就递归查找左区间， 如果要查找的区间在节点的又间里面，那么就递归查找右区间，如果要查找的区间既有左区间部分也有右区间部分，我们就在左区间和右区间的返回值里面找最大值。注意每次写函数最好是先写边界，有助于理解。

int querymax(int id,int ql,int qr)
{
    int l=tree[id].l;
    int r=tree[id].r;
    if(l>=ql&&r<=qr)
        return tree[id].maxans;
    int mid=(l+r)/2;
    if(mid>=qr)
        return querymax(id*2,ql,qr);
    else if(ql>=(mid+1))
        return querymax(id*2+1,ql,qr);
    else
        return max(querymax(id*2,ql,qr),querymax(id*2+1,ql,qr));
}

最好贴上完整的代码：

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define maxn 200005
int a[maxn];
struct node
{
    int id;
    int l;
    int r;
    int maxans;
}tree[4*maxn];
void update(int id,int x,int newval)
{
    int l=tree[id].l;
    int r=tree[id].r;
    int mid=(l+r)/2;
    if(l==r)
    {
        tree[id].maxans=newval;
        return;
    }
    if(x<=mid)
        update(id*2,x,newval);
    else
        update(id*2+1,x,newval);
    tree[id].maxans=max(tree[id*2].maxans,tree[id*2+1].maxans);
}
int querymax(int id,int ql,int qr)
{
    int l=tree[id].l;
    int r=tree[id].r;
    if(l>=ql&&r<=qr)
        return tree[id].maxans;
    int mid=(l+r)/2;
    if(mid>=qr)
        return querymax(id*2,ql,qr);
    else if(ql>=(mid+1))
        return querymax(id*2+1,ql,qr);
    else
        return max(querymax(id*2,ql,qr),querymax(id*2+1,ql,qr));
}
void build(int id,int l,int r)
{
    tree[id].l=l;
    tree[id].r=r;
    if(l==r)
    {
        tree[id].maxans=a[l];
        return;
    }
    int mid=(l+r)/2;
    build(id*2,l,mid);
    build(id*2+1,mid+1,r);
    tree[id].maxans=max(tree[id*2].maxans,tree[id*2+1].maxans);
}
int main()
{
    int n,q;
    char s[10];
    while(scanf("%d%d",&n,&q)!=EOF)
    {
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]);
        build(1,1,n);
        while(q--)
        {
            int c,b;
            scanf("%s%d%d",s,&c,&b);
            if(s[0]=='U')
                update(1,c,b);
            else
            {
                int ans=querymax(1,c,b);
                printf("%d\n",ans);
            }
        }
    }
return 0;
}

第一次写线段树，还有很多小细节要问题，建树时的更新，查找时对区间的处理，代码能力还是太弱。。。。。