[计蒜客-树状数组线段树]帕吉的肉钩

分析题意, 是一道典型的线段树题目.

要求做到区间赋值(树状数组应该无法做到), 区间求和两个操作.

开始的时候思路不清楚, 后来要注意线段树的 up 和 down 操作.

需要注意的是有lazy-tag的节点, 它本身的信息是完全正确的, 子节点的信息是待更新的.

这也就是说, 当需要用到这些子节点的信息时在进行更新.

#include <iostream>
#include <algorithm>
#include <cstdlib>
#include <cstdio>
#include <cmath>
using namespace std;
const int MAXN = 1e5 + 600;
int s[MAXN << 2 | 1];
int flag[MAXN << 2 | 1];
int n;
int ql, qh, v;
inline void up(int o){
    s[o] = s[o << 1] + s[o << 1 | 1];
}
inline void down(int o, int lo, int hi){
    if(flag[o]){
        flag[o << 1] = flag[o << 1 | 1] = flag[o];
        int mi = lo + (hi - lo >> 1);
        s[o << 1] = (mi - lo) * flag[o];
        s[o << 1 | 1] = (hi - mi) * flag[o];
        flag[o] = 0;
    }
}
void build(int o, int lo, int hi){
    if(hi - lo < 2){
        s[o] = 1;
        return;
    }
    int mi = lo + (hi - lo >> 1);
    build(o << 1, lo, mi);
    build(o << 1 | 1, mi, hi);
    up(o);
}
int query(int o, int lo, int hi){
    if(ql <= lo && hi <= qh){
        return s[o];
    }
    if(qh <= lo || hi <= ql)
        return 0;

    down(o, lo, hi);
    int rst = 0, mi = lo + (hi - lo >> 1);

    rst += query(o << 1, lo, mi);
    rst += query(o << 1 | 1, mi, hi);
    return rst;
}
void modify(int o, int lo, int hi){
    if(ql <= lo && hi <= qh){
        flag[o] = v;
        s[o] = (hi - lo) * v;
        return;
    }
    if(qh <= lo || hi <= ql)
        return;

    down(o, lo, hi);
    int mi = lo + (hi - lo >> 1);
    modify(o << 1, lo, mi);
    modify(o << 1 | 1, mi, hi);

    up(o);
}
int main(){
    scanf("%d", &n);
    build(1, 1, ++n);
    int q;
    scanf("%d", &q);
    while(q--){
        scanf("%d%d%d", &ql, &qh, &v);
        ++qh;
        modify(1, 1, n);
    }
    ql = 1, qh = n;
    printf("The total value of the hook is %d.\n", s[1]);
    return 0;
}