hdu1754 I hate it (线段树)

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<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

Author
linle

分析:线段树入门基础题,每次更新结点存储该以结点为根的子树中最大值。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <cmath>
#include <algorithm>
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;
}


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