HDU1754-I Hate It

典型的线段树问题，坑点在于有多组测试数据…

#include <cstdio>
#include <algorithm>

using namespace std;

const int maxn = 1 << 18;

int n, dat[2*maxn-1];

void init(int n_) {
    n = 1;
    while (n < n_) {
        n *= 2;
    }
    for (int i = 0; i < 2 * n - 1; i++) {
        dat[i] = 0;
    }
}

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 0;
    }
    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 const *argv[]) {
    int N, M;
    while (scanf("%d%d", &N, &M) == 2) {
        init(N);
        for (int i = 0; i < N; i++) {
            int mark;
            scanf("%d", &mark);
            update(i, mark);
        }
        getchar();
        for (int i = 0; i < M; i++) {
            char ch;
            int a, b;
            scanf("%c%d%d", &ch, &a, &b);
            getchar();
            a--;
            if (ch == 'Q') {
                printf("%d\n", query(a, b, 0, 0, n));
            } else {
                update(a, b);
            }
        }
    }
    return 0;
}