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 + ;
 int sum[MAXN << ];
 int col[MAXN << ];

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

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

 void build(int l, int r, int rt){ //建树
     col[rt] = ;
     sum[rt] = ;
     if(l == r) return;
     int mid = (l + r) >> ;
     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 + ) * key;
         return;
     }
     push_down(rt, r - l + );//向下更新
     int  mid = (l + r) >> ;
     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 = ; i <= t; i++){
         scanf("%d%d", &n, &q);
         build(, n, );
         while(q--){
             scanf("%d%d%d", &x, &y, &z);
             update(x, y, z, , n, );
         }
         printf("Case %d: The total value of the hook is %d.\n", i, sum[]);//求的是整个数组的和,所以不需要另外写query函数
     }
     return ;
 }