hdu 1754 I Hate It

题目链接：hdu 1754 I Hate It 

线段树模板题。该题是对点进行更新，对区间进行查询，在初始化时为了方便把n扩充为了2的整数次幂。并且注意线段树应该开辟4倍于n的数组来存储。注意对于输入字符'Q','U'，一定不能用char c;scanf("%c",&c);来接收，这样会接收到上次输入的回车。正确的方法是使用char s[3];scanf("%s",s);，这个bug我找了好久才发现。

代码如下：

#include <cstdlib>
#include  <cstdio>
#include  <iostream>
#include  <cstring>
using namespace std;
const int MAX_N = 1 << 20 ;
int n, dat[MAX_N * 2 -1];
const int MIN = -1;
void init(int n_)
{
    n = 1;
    while(n < n_) n*=2;
    for( int i = 0 ; i < 2 * n -1 ; i++ )
    {
        dat[i] = MIN;
    }
}
void update(int k, int a)
{
    k += n - 1;
    dat[k] = a;
    while( k > 0 )
    {
        k = (k-1)/2;
        dat[k] = max(dat[k*2+1], dat[k*2+2]);
    }
}
int query(int a, int b, int k, int l, int r)
{
    if( r <= a || b <= l ) return MIN;
    if( a <= l && r <= b ) return dat[k];
    else
    {
        int vl = query(a, b, k * 2 + 1, l, (l+r)/2);
        int vr = query(a, b, k * 2 + 2, (l+r)/2, r);
        return max(vl, vr);
    }
}
int main(int argc, char *argv[])
{
    int N, M;
    while(scanf("%d%d", &N,&M ) != EOF)
    {
        int tmp;
        init(N);
        for( int i = 0 ; i < N ; i++ )
        {
            scanf ( "%d", &tmp );
            update(i,tmp); 
        }
        char c[5];
        for( int i = 0 ; i < M ; i++ )
        {
            scanf ( "%s", c );
            if(!strcmp(c ,"Q"))
            {
                int l, r;
                scanf ( "%d%d", &l, &r);
                printf("%d
", query(l-1, r, 0, 0, n));
            }
            else if( !strcmp(c , "U" ))
            {
                int a, b;
                scanf ( "%d%d", &a, &b );
                update(a-1, b);
            }
        }
    }
}