线段树解决HDOJ 1754

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

 

 

 

一次AC，感觉不错哟!嘿嘿！

另外，在POJ上也有这道题目，但是，

在POJ上的时间好像是1000ms,我的

是2031ms,所以我的代码在那里会超

时的。有很多优化的方法，自己上网

搜索去吧。

#include "iostream"
using namespace std;
struct Node
{
    int left, right, mid, score;
};
Node Tree[600000];
void BuildTree(int level, int left, int right){  //建树
    Tree[level].score = -1;
    Tree[level].left = left;
    Tree[level].right = right;
    Tree[level].mid = (left+right)/2;
    if(left==right){
        return;
    }
    else{
        BuildTree(2*level, left, Tree[level].mid);
        BuildTree(2*level+1, Tree[level].mid+1, right);
    }
}
void Insert(int level, int stuID, int score){
    if(stuID==Tree[level].left && stuID==Tree[level].right){  //递归的结束条件
        Tree[level].score = score;
        return;
    }
    if(stuID>Tree[level].mid){  //往右边搜索
        if(score>Tree[level].score)  //搜索过程中更新最大值
            Tree[level].score = score;
        Insert(2*level+1, stuID, score);
    }
    else if(stuID<=Tree[level].mid){  //往左边搜索
        if(score>Tree[level].score)  //搜索过程中更新最大值
            Tree[level].score = score;
        Insert(2*level, stuID, score);
    }
}
int Find(int level, int left, int right){
    if(left==Tree[level].left && right==Tree[level].right){
        return Tree[level].score;
    }
    else if(left > Tree[level].mid){  //向右边搜索
        Find(2*level+1, left, right);
    }
    else if(right <= Tree[level].mid){  //向左边搜索
        Find(2*level, left, right);
    }
    else{   //向两边搜索，取较这边中最大数的较大者
        int a=Find(2*level, left, Tree[level].mid);
        int b=Find(2*level+1, Tree[level].mid+1, right);
        return a>b?a:b;
    }
}
int main()
{
    int numofstu, numofoper;
    while(cin>>numofstu>>numofoper){
        BuildTree(1, 1, 200000);
        int score, stuID, left, right;
        char op;
        for(int i=1; i<=numofstu; i++){
            cin>>score;
            Insert(1, i, score);
        }
        while(numofoper--){
            cin>>op;
            if(op=='Q'){
                cin>>left>>right;
                cout<<Find(1, left, right)<<endl;
            }
            else{
                cin>>stuID>>score;
                Insert(1, stuID, score);
            }
        }
    }
}