HDU 1698 Just a Hook 线段树区间修改

最新推荐文章于 2022-04-01 12:15:40 发布

原创最新推荐文章于 2022-04-01 12:15:40 发布 · 266 阅读

1 ·

CC 4.0 BY-SA版权

数据结构_线段树专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一道经典的线段树区间修改问题，通过给定的一系列操作，更新指定区间的数值并最终求出整个区间的和。文章提供了完整的C++代码实现，详细展示了线段树构建、下推及更新过程。

传送门：点击打开链接

题意：
给定N个数，初始化全部为1.
一种操作，区间修改。把区间[L,R]内所有数字修改为另一个值X
Q次操作之后，求区间[1,N]的和

分析：
简单的线段树区间修改操作应用题
具体实现见代码

#include <cstdio>
using namespace std;

const int maxn = 100000+5;
struct node{
   int left,right;
   int lazy,sum;
};

struct Tree{
   node tr[maxn*4];
   #define lch(u) (u<<1)
   #define rch(u) (u<<1|1)
   #define Mid (tr[u].left+tr[u].right)>>1

   inline void push_up(int u){ tr[u].sum = tr[lch(u)].sum + tr[rch(u)].sum; }

   void build( int u, int a, int b){
      tr[u].left = a; tr[u].right = b; tr[u].lazy = 0;
      if (a == b) {
         tr[u].sum = 1;
         return;
      }
      int mid = Mid;
      build(lch(u),a,mid);
      build(rch(u),mid+1,b);
      push_up(u);
   }

   void push_down(int u){
       if (tr[u].lazy == 0) return;
       tr[lch(u)].lazy = tr[rch(u)].lazy = tr[u].lazy;
       int mid  = Mid;
       tr[lch(u)].sum = (mid-tr[u].left+1)*tr[u].lazy;
       tr[rch(u)].sum = (tr[u].right-mid)*tr[u].lazy;
       tr[u].lazy = 0;
   }

   void update(int u, int a, int b, int val) {
       if (tr[u].left == a && tr[u].right == b) {
           tr[u].lazy = val;
           tr[u].sum = (b-a+1)*val;
           return;
       }

       push_down(u);

       int mid = Mid;
       if (b<=mid) update(lch(u),a,b,val);
       else if (a>mid) update(rch(u),a,b,val);
       else {
           update(lch(u),a,mid,val);
           update(rch(u),mid+1,b,val);
       }
       push_up(u);
   }


}T;
int n,q;
int x,y,z;

int main(){
    int t;
    scanf("%d",&t);

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