HDU 1698 Just a Hook // 线段树 区间更新

题目描述

HDU 1698 Just a Hook

解题思路

题目大意:
每段区间有三类状态: 值为1, 2, 3
要求支持下列操作:
1) 修改区间的状态, 然后输出整个区间[1, n]的和
线段树区间更新的入门题, 具体看注释哈~

参考代码

#include <stdio.h>
#define lson rt<<1
#define rson rt<<1|1
#define mid ((l+r) >> 1)
const int MAX_N = 100010;
struct SegTree{
    int sum, kind; // sum是区间的和, kind是区间的状态(也是延迟标记)
}node[MAX_N << 2];

// 更新根节点
void pushup(int rt){ 
    node[rt].sum = node[lson].sum + node[rson].sum;
}

// 延迟更新
void pushdown(int rt, int len){
    if (node[rt].kind){ // 根节点有延迟标记
         // 向左右儿子传递标记
        node[lson].kind = node[rson].kind = node[rt].kind;  
        // 左儿子的值更新为 左儿子区间的长度*状态      
        node[lson].sum = (len - (len >> 1)) * node[rt].kind;
        // 右儿子的值更新为 右儿子区间的长度*状态   
        node[rson].sum = (len >> 1) * node[rt].kind;
        // 去掉根节点的延迟标记
        node[rt].kind = 0;
    }
}

void build(int l, int r, int rt){
    node[rt].kind = 0; //开始是没有延迟标记的
    node[rt].sum = 1; // 由题知,初始状态的值为1
    if (l == r) return ;
    build(l, mid, lson);
    build(mid+1, r, rson);
    pushup(rt);
}

void update(int L, int R, int kind, int l, int r, int rt){
    int len = r - l + 1; // 区间[l, r]的长度
    if (L <= l && r <= R) // 区间[L, R] 完全包含 区间[l, r]
        node[rt].kind = kind; 更新状态
        node[rt].sum = kind * len; 更新值
        return ;
    }
    pushdown(rt, len); // 释放延迟标记
    if (L <= mid)   update(L, R, kind, l, mid, lson);
    if (R > mid)    update(L, R, kind, mid+1, r, rson);
    pushup(rt);
}

int main(){
    int T, n, m, x, y, z, kase = 1;
    scanf("%d", &T);
    while (T--){
        scanf("%d %d", &n, &m);
        build(1, n, 1);
        while (m--){
            scanf("%d %d %d", &x, &y, &z);
            update(x, y, z, 1, n, 1);
        }
        printf("Case %d: The total value of the hook is %d.\n", kase++, node[1].sum);
    }
    return 0;
}