ccnu-线段树联系-单点更新2-B

B - 单点更新2
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
对于每一次询问操作,在一行里面输出最高成绩。

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
Hint
Huge input,the C function scanf() will work better than cin 

代码如下:
#include<iostream>
#include<string>
#include<cstdio>

#define Max 200005
#define INF -1
using namespace std;
struct BTm
{
    int elem;
    int r;
    int l;
};

BTm an[Max<<3];
int stu[Max];
int max(int a,int b)
{
    return a>b?a:b;
}
void init(int i,int j, int k)
{
    an[k].r = j;
    an[k].l = i;
    if(i == j)
    {
        an[k].elem = stu[i];
       // printf("z\n");
        return ;
    }
    int m=(i+j)>>1;
    init(i,m,k<<1);
    init(m+1,j,k<<1|1);
    an[k].elem = max(an[k<<1].elem , an[k<<1|1].elem);

}

void update(int index,int change,int k)
{
     int r,l;
     r = an[k].r;
     l = an[k].l;
     if(r == l) {an[k].elem = change; return ;}
     int m = (r+l)>>1;

     if(index <= m)
       update(index,change,k<<1);
     else
       update(index,change,k<<1|1);
    an[k].elem = max(an[k<<1].elem ,an[k<<1|1].elem);
}
int query(int lstu,int rstu,int k)
{
    int l = an[k].l;
    int r = an[k].r;
    if(lstu == l&&rstu == r)
    {
        return an[k].elem;
    }
    int ans = -1;
    int m = (l + r)>>1;
    if(rstu <= m)
    {
        ans = max(query(lstu,rstu,k<<1),ans);
    }
    else if(m < lstu)
    {
        ans = max(query(lstu,rstu,k<<1|1),ans);
    }
    else
    {
        ans = max(ans,max(query(lstu,m,k<<1),query(m+1,rstu,k<<1|1)));
    }
    return ans;
}
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m) != EOF)
{
    for(int i = 1; i <= n; i++)
      scanf("%d",&stu[i]);
    init(1,n,1);
    while(m--)
    {
        string s;
        cin>>s;
        if(s == "Q")
        {
            int A,B;
            scanf("%d%d",&A,&B);
            printf("%d\n",query(A,B,1));
        }
        else
        {
            int A,B;
            scanf("%d%d",&A,&B);
            update(A,B,1);
        }
 }   }
    return 0;
}
【位运算符一定是可耐的男孩纸!*^*】

你可能感兴趣的:(线段树)