[計蒜客-樹狀數組線段樹]帕吉的肉鉤

分析題意, 是一道典型的線段樹題目.

要求做到區間賦值(樹狀數組應該無法做到), 區間求和兩個操作.

開始的時候思路不清楚, 後來要注意線段樹的 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;
}