I Hate It（基本线段树）

I Hate It

Time Limit : 9000/3000ms (Java/Other)   Memory Limit : 32768/32768K (Java/Other)

Total Submission(s) : 24   Accepted Submission(s) : 7

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[/hint]
  
  

 

Author

linle

 

Source

2007省赛集训队练习赛（6）_linle专场

直接用线段树求区间最大值及更新值即可，但是我写的代码一直没能AC求指教。。。

正确代码：”

#include<iostream>
#include<string>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<iomanip>
#include<queue>
#include<cstring>
#include<map>
using namespace std;
typedef long long ll;
#define M 200000
int n,m,a[M];
struct node{
    int l,r,maxn; //maxn直接用来存区间内的最大值
}tree[M*4+5];
void build(int id,int l,int r)//id是节点编号，l,r是当前节点所包含的区间范围
{
    tree[id].l=l; tree[id].r=r; //建立第id个节点，它管理区间为l-r
    if(l==r)//如果是叶子节点，直接输入数据并返回
    {
        scanf("%d",&tree[id].maxn);
        return;
    }
    else  //如果不是叶子节点
    {
        int mid=(l+r)>>1; //mid为它区间中点
        build(id<<1,l,mid);//当mid小于等于左边区间建左子树
        build(id<<1|1,mid+1,r);//建右子树
        tree[id].maxn=max(tree[id<<1].maxn,tree[id<<1|1].maxn);//更新区间最大值
    }
}
void update(int id,int pos,int val)//id是节点编号，pos是要更新的点
{
    if(tree[id].l==tree[id].r)//找到要更新的点，赋值并返回
    {
        tree[id].maxn=val;
        return;
    }
    else
    {
        int mid=(tree[id].l+tree[id].r)>>1;//mid为节点区间中点
        if(pos<=mid) update(id<<1,pos,val);//点的位置在当前节点管理区间左半部分，
        else update(id<<1|1,pos,val); //点的位置在当前节点管理区间右半部分，
        tree[id].maxn=max(tree[id<<1].maxn,tree[id<<1|1].maxn);
    }
}
int query(int id,int l,int r)//id是节点编号,l,r是要查询区间
{
    if(tree[id].l==l&&tree[id].r==r) //找到区间
    {
        return tree[id].maxn;
    }
    else
    {
        int mid=(tree[id].l+tree[id].r)>>1; //mid是当前节点管理区间的中点
        if(r<=mid) return query(id<<1,l,r); //要查询区间完全在左子树上
        else if(l>mid) return query(id<<1|1,l,r);//要查询区间完全在右子树上
        else return max(query(id<<1,l,mid),query(id<<1|1,mid+1,r)); //要查询区间分为两部分，取其中最大的一个
    }
}
int main()
{
    int left,right;
    char ch;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        build(1,1,n);
        while(m--)
        {
            getchar();
            ch=getchar();
            scanf("%d%d",&left,&right);
            if(ch=='Q')
            {
                printf("%d\n",query(1,left,right));
            }
            else
            {
                update(1,left,right);
            }
        }
    }
    return 0;
}

下面附上错误代码（wrong answer）：

#include<iostream>
#include<string>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<iomanip>
#include<queue>
#include<cstring>
#include<map>
using namespace std;
typedef long long ll;
#define M 200000
int n,m,a[M];
struct node{
    int l,r,maxn; 
}tree[M*4+5];
void build(int id,int l,int r)
{
    tree[id].l=l; tree[id].r=r; 
    if(l==r)
    {
        scanf("%d",&tree[id].maxn);
        return;
    }
    else
    {
        int mid=(l+r)>>1; 
        build(id<<1,l,mid);
        build(id<<1|1,mid+1,r);
        tree[id].maxn=max(tree[id<<1].maxn,tree[id<<1|1].maxn);
    }
}
void update(int id,int pos,int val)
{
    if(tree[id].l==tree[id].r)
    {
        tree[id].maxn=val;
        return;
    }
    else
    {
        int mid=(tree[id].l+tree[id].r)>>1;
        if(pos<=mid) update(id<<1,pos,val);
        else update(id<<1|1,pos,val);
        tree[id].maxn=max(tree[id<<1].maxn,tree[id<<1|1].maxn);
    }
}
int query(int id,int l,int r)
{
    if(tree[id].l==l&&tree[id].r==r) 
    {
        return tree[id].maxn;
    }
    else
    {
        int mid=(tree[id].l+tree[id].r)>>1;
        if(r<=mid) return query(id<<1,l,r); 
        else if(l>mid) return query(id<<1|1,l,r);
        else return max(query(id<<1,l,mid),query(id<<1|1,mid+1,r));
    }
}
int main()
{
    int left,right;
    char ch;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        build(1,1,n);
        while(m--)
        {
            getchar();
            ch=getchar();
            scanf("%d%d",&left,&right);
            if(ch=='Q')
            {
                printf("%d\n",query(1,left,right));
            }
            else
            {
                update(1,left,right);
            }
        }
    }
    return 0;
}