程式師世界 >> 編程語言 >> C語言 >> C++ >> 關於C++ >> hdu1754 I hate it （線段樹）

hdu1754 I hate it （線段樹）

編輯：關於C++

I Hate It

Time Limit: 9000/3000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others) Total Submission(s): 55291 Accepted Submission(s): 21599

Problem Description 很多學校流行一種比較的習慣。老師們很喜歡詢問，從某某到某某當中，分數最高的是多少。
這讓很多學生很反感。

不管你喜不喜歡，現在需要你做的是，就是按照老師的要求，寫一個程序，模擬老師的詢問。當然，老師有時候需要更新某位同學的成績。
Input 本題目包含多組測試，請處理到文件結束。
在每個測試的第一行，有兩個正整數 N 和 M ( 0 學生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

Sample Output

5
6
5
9

HintHuge input,the C function scanf() will work better than cin

Author linle
分析：線段樹入門基礎題，每次更新結點存儲該以結點為根的子樹中最大值。

#include 
#include 
#include 
#include 
#include 
#include
#include 
#include 
#include 
#include 
using namespace std;
const double eps = 1e-6;
const double pi = acos(-1.0);
const int INF = 0x3f3f3f3f;
const int MOD = 1000000007;
#define ll long long
#define CL(a,b) memset(a,b,sizeof(a))
#define MAXN 800010

struct node
{
    int a,b,r;
}t[MAXN];
int n,m,ans;

void build(int x, int y, int num)
{
    t[num].a = x;
    t[num].b = y;
    t[num].r = 0;
    if(x == y) return ;
    int mid = (x+y)/2;
    build(x, mid, num*2);
    build(mid+1, y, num*2+1);
}

void update(int x, int y, int num)
{
    if (t[num].a == t[num].b && t[num].b == x)
    {
        t[num].r = y;
        return ;
    }
    int mid = (t[num].a+t[num].b)/2;
    if(x > mid) update(x, y, num*2+1);
    else update(x, y, num*2);
    t[num].r = max(t[num*2].r, t[num*2+1].r);
}

void query(int x, int y, int num)
{
     if(t[num].a == x && t[num].b == y)
     {
         ans = max(ans, t[num].r);
         return ;
     }
     int mid = (t[num].a+t[num].b)/2;
     if(x >= mid+1) query(x, y, num*2+1);
     else if(y <= mid) query(x, y, num*2);
     else
     {
         query(x, mid, num*2);
         query(mid+1, y, num*2+1);
     }
}

int main()
{
    char ch;
    int x,y,k;
    while(scanf("%d%d",&n,&m)==2)
    {
        build(1, n, 1);
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&k);
            update(i, k, 1);
        }

        while(m--)
        {
            getchar();
            scanf("%c%d%d",&ch,&x,&y);
            if(ch == 'U')
            {
                update(x, y, 1);
            }
            else if(ch == 'Q')
            {
                ans = -99999;
                query(x, y, 1);
                printf("%d\n",ans);
            }
        }
    }
    return 0;
}