HDU 1754 – I Hate It [线段树]

突然发现自己落伍了,最近几个月很多算法都忽略了,近期准备学习下线段树,最小生成树还有最短路问题和赛瓦定理。

这道题应该算是很简单的线段树模板题了,参考《挑战程序设计竞赛》做的,貌似还有其他线段树的做法

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int maxn = 1 << 21;
int n, dat[maxn];

void init(int n_){
    n = 1;
    while(n < n_) n <<= 1;
    memset(dat, -1, sizeof(dat));
}
void update(int k, int a){
    k += n - 1; dat[k] = a;
    while(k > 0){
        k = (k - 1) / 2;
        dat[k] = max(dat[2*k + 1], dat[2*k + 2]);
    }
}
int query(int a, int b, int k, int l, int r){
    if(r <= a || l >= b) return -1;
    if(a <= l && r <= b) return dat[k];
    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(){
    ios::sync_with_stdio(false);
    int N, M;
    while(cin >> N >> M){
        init(N);
        for(int i = 0; i < N; ++i){
            int x; cin >> x;
            update(i, x);
        }
        for(int i = 0; i < M; ++i){
            char c; int a, b; cin >> c >> a >> b;
            if(c == 'Q') printf("%d\n", query(a-1, b, 0, 0, n));
            if(c == 'U') update(a-1, b);
        }
    }
    return 0;
}

 

欢迎留言