HDU 1698 Just a Hook（线段树区间替换）

最新推荐文章于 2019-07-03 21:31:19 发布

mirror58229

最新推荐文章于 2019-07-03 21:31:19 发布

阅读量281

点赞数

CC 4.0 BY-SA版权

分类专栏：数据结构

本文链接：https://blog.youkuaiyun.com/mirror58229/article/details/53870672

数据结构专栏收录该内容

18 篇文章

订阅专栏

本文介绍了一种基于线段树的数据结构实现方法，用于高效地处理区间元素的批量替换操作。通过递归构建线段树，并利用懒惰传播（lazy propagation）优化更新操作，使得在大规模数据集上进行区间替换变得更为高效。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：把指定区间替换成1 2 3
修改一下update与pushDown，直接把add替换

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
typedef long long ll;
const int N=1e5+10;
struct Node{
    ll l,r;
    ll sum,add;
}tree[4*N];
void build(ll i,ll l,ll r){
    tree[i].l=l,tree[i].r=r;
    tree[i].add=0;
    if(l==r){//左边=右边，单点建立
        tree[i].sum=1;
        return;
    }
    ll tmp=i<<1;
    ll mid=(l+r)>>1;
    build(tmp,l,mid);//左子树
    build(tmp+1,mid+1,r);//右子树
    tree[i].sum=tree[tmp].sum+tree[tmp+1].sum;//回溯得到区间sum值
}
void pushDown(ll x){
    ll tmp=x<<1;
    tree[tmp].add=tree[x].add;//左子树得到父节点标记
    tree[tmp+1].add=tree[x].add;//右子树得到父节点标记

    //如果区间内每个元素变成一个值,那么这个区间会变成当前区间包含指定区间元素个数*待加元素的值
    tree[tmp].sum=tree[x].add*(tree[tmp].r-tree[tmp].l+1);
    tree[tmp+1].sum=tree[x].add*(tree[tmp+1].r-tree[tmp+1].l+1);
    tree[x].add=0;//清空父节点标记
}

void update(ll i,ll l,ll r,ll s){
    if(tree[i].l>r||tree[i].r<l){//超出范围 
        return ;
    }

    //在指定区间的范围内,该点加上sum与延迟标记.此时包含该点的父区间均更新完毕 
    //>=、<=是考虑到区间位于不同子树的情况 

    if(tree[i].l>=l&&tree[i].r<=r){
        tree[i].add=s;
        tree[i].sum=s*(tree[i].r-tree[i].l+1);
        return;
    }
    //如果之前已经有延迟标记,需要将之前的标记处理 
    if(tree[i].add) pushDown(i); 
    ll tmp=i<<1;

    //更新左右子树,如果某个子树超出了所更新区间的范围,会在第一个条件处return 

    update(tmp,l,r,s);
    update(tmp+1,l,r,s);

    tree[i].sum=tree[tmp].sum+tree[tmp+1].sum;//回溯更新父节点的值   
}

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

}