hdu1698(线段树区间替换模板)

题目链接: http://acm.hdu.edu.cn/showproblem.php?pid=1698

题意: 第一行输入 t 表 t 组测试数据, 对于每组测试数据, 第一行输入一个 n , 表示钩子有 n 节, 编号为 1 ~ n, 每节钩子的初始价值为 1 , 接下来输入一个 q,

接着 q 行输入, 每行格式为 l, r, x, 表示讲区间 [l, r] 内的钩子价值变成 x , 求最终的总价值；

思路: 线段树区间替换模板

代码:

#include <iostream>
#include <stdio.h>
#define lson l, mid, rt << 1
#define rson mid + 1, r, rt << 1 | 1
using namespace std;

const int MAXN = 1e5 + 10;
int sum[MAXN << 2];
int col[MAXN << 2];

void push_up(int rt){//向上更新
    sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];
}

void push_down(int rt, int m){
    if(col[rt]){//若有标记,则将标记移向下一层
        col[rt << 1] = col[rt << 1 | 1] = col[rt]; //将标记向下更新
        sum[rt << 1] = (m - (m >> 1)) * col[rt];//更新左儿子
        sum[rt << 1 | 1] = (m >> 1) * col[rt];//更新右儿子
        col[rt] = 0; //已经更新过的取消标记
    }
}

void build(int l, int r, int rt){ //建树
    col[rt] = 0;
    sum[rt] = 1;
    if(l == r) return;
    int mid = (l + r) >> 1;
    build(lson);
    build(rson);
    push_up(rt);
}

void update(int L, int R, int key, int l, int r, int rt){ //区间替换
    if(L <= l && R >= r){
        col[rt] = key;//延时标记
        sum[rt] = (r - l + 1) * key;
        return;
    }
    push_down(rt, r - l + 1);//向下更新
    int  mid = (l + r) >> 1;
    if(L <= mid) update(L, R, key, lson);
    if(R > mid) update(L, R, key, rson);
    push_up(rt);//向上更新
}

int main(void){
    int t, n, q, x, y, z;
    scanf("%d", &t);
    for(int i = 1; i <= t; i++){
        scanf("%d%d", &n, &q);
        build(1, n, 1);
        while(q--){
            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.
", i, sum[1]);//求的是整个数组的和,所以不需要另外写query函数
    }
    return 0;
}