线段树解决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,所以我的代码在那里会超

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

搜索去吧。

View Code

#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);

            }

        }

    }

}