【HDU 1754】I Hate It —— 线段树（数组和指针实现）

原题链接

I Hate It

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

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
   
   
  
  

解题报告：

由于题目要求多次查询和更新数据，所以用简单的数组查询肯定会TLE。因而改用统计方法来实现，最经典的肯定是使用线段树来实现。建树，区间查询，点更新。这么简单的题一直TLE，最后发现竟然是没加 !=EOF ，一直停留在while里面，摔！！！

 /*
  *线段树的指针实现
  */
#include <iostream>
#include <cstdio>
using namespace std;
int n,m,l,r; char c[10];
const int maxn=200010;
int stu[maxn];
struct Node{
    int a,b,MAX;
    Node* l;Node* r;
    Node(int x,int y):a(x),b(y),l(NULL),r(NULL){}
}*root;

inline int max(int a,int b){return a>b?a:b;}
void build(int a,int b,Node*& rt)
{
    if(b==a) {rt=new Node(a,b);rt->MAX=stu[a];return;}

    rt = new Node(a,b);
    int mid=(b-a+1)/2+a-1;
    build(a,mid,rt->l);
    build(mid+1,b,rt->r);
    rt->MAX = max(rt->l->MAX,rt->r->MAX);
}
int query(int a,int b,Node* rt)
{
    if(a == rt->a && b== rt->b) return rt->MAX;

    int mid=rt->l->b;
    if(a<=mid && b>mid)
        return max( query(a,mid,rt->l),query(mid+1,b,rt->r) );
    else{
        if(b <= mid) return query(a,b,rt->l);
        else return query(a,b,rt->r);
    }
}
void update(const int& a,const int& score,Node* rt)
{
    if(a == rt->a && a== rt->b) {
        rt->MAX = score;
    }
    else{
        if(a<= rt->l->b) update(a,score,rt->l);
        else update(a,score,rt->r);

        rt->MAX = max(rt->l->MAX,rt->r->MAX);
    }

}
void destroy(Node* rt)
{
    if(rt->l == NULL) delete rt;
    else {
        destroy(rt->l);
        destroy(rt->r);
        delete rt;
    }
}
int main()
{
    while( scanf("%d%d",&n,&m)!=EOF )
    {
        for(int i=1;i<=n;++i) scanf("%d",&stu[i]);
        build(1,n,root);

        for(int i=0;i<m;++i){
            scanf("%s%d%d",c,&l,&r);
            if(c[0]=='Q') printf( "%d\n",query(l,r,root) );
            else if(c[0]=='U')update(l,r,root);
        }
        destroy(root); //记得要delete掉，要不然会MLE
    }
    return 0;
}

 /*
  *线段树的数组实现
  */
#include <cstdio>
#include <iostream>
using namespace std;

struct Node{
    int L,R,MAX;
    void set(int a,int b,int c=-1){L=a;R=b;MAX=c;}
};
int n,m,a,b; char c[5];
const int maxn=200010;
int stu[maxn];
Node segTree[4*maxn];

inline int max(int a,int b){return a>b?a:b;}
inline int lchild(int p){return  p<<1;   }
inline int rchild(int p){return (p<<1)|1;}
void build_segment_tree(int cur,int l,int r)
{
    if(l==r) {segTree[cur].set(l,r,stu[l]);return;}

    segTree[cur].set(l,r);
    build_segment_tree(lchild(cur), l ,  (l+r)>>1);
    build_segment_tree(rchild(cur),((l+r)>>1)+1,r);
    // 如果max是宏定义，不要直接把函数作为参数，否则会多次调用
    segTree[cur].MAX=max(segTree[lchild(cur)].MAX,segTree[rchild(cur)].MAX);
}
int query(int cur,const int& l,const int& r)
{   if(r<segTree[cur].L || segTree[cur].R<l) return -1;
    if(l<=segTree[cur].L && segTree[cur].R<=r) return segTree[cur].MAX;

    // 如果max是宏定义，不要直接把函数作为参数，否则会多次调用
    int mid=(segTree[cur].L+segTree[cur].R)>>1, ans=-1;
    if(l<=mid)ans=max(ans,query(lchild(cur),l,r));
    if(mid<r) ans=max(ans,query(rchild(cur),l,r));
    return ans;
}
void update(int cur,const int& s,const int& score)
{
    if(segTree[cur].L==segTree[cur].R) {segTree[cur].MAX=score;return;}

    if(s<= (segTree[cur].L+segTree[cur].R)>>1 ) update(lchild(cur),s,score);
    else update(rchild(cur),s,score);
    // 如果max是宏定义，不要直接把函数作为参数，否则会多次调用
    segTree[cur].MAX=max(segTree[lchild(cur)].MAX, segTree[rchild(cur)].MAX);
}

int main()
{
    while( cin>>n>>m )
    {
        for(int i=1;i<=n;++i) scanf("%d",&stu[i]);
        build_segment_tree(1,1,n);

        for(int i=0;i<m;++i){
            scanf("%s%d%d",c,&a,&b);
            if(c[0]=='Q') {printf( "%d\n",query(1,a,b) );}
            else update(1,a,b);
        }
    }
    return 0;
}