hdu 1698 线段树区间更新入门(lazy标记)

点击打开链接

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
using namespace std;
const int M=1e5+20;
long long sum[M*4],lazy[M*4];//lazy标记 
struct node{
	int l,r;
	int mid()
	{
		return (l+r)>>1;
	}
	
}seg[M<<2];
void pushup(int rt)
{
	sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int l,int r,int rt)
{
	seg[rt].l=l;
	seg[rt].r=r;
	lazy[rt]=0;
	if(l==r)
	{
		sum[rt]=1;//初始化 
		return;
	}
	int m=seg[rt].mid();
	build(l,m,rt<<1);
	build(m+1,r,rt<<1|1);
	
	pushup(rt);

}
void Pushdown(int rt,int l)
{
	if(lazy[rt])
	{
		lazy[rt<<1]=lazy[rt];//传递lazy标记
		lazy[rt<<1|1]=lazy[rt];
		
		sum[rt<<1]=lazy[rt]*(l-l/2);
		sum[rt<<1|1]=lazy[rt]*(l/2);
		
		lazy[rt]=0;//更新后取消lazy标记 
		
	}
}
void update(int c,int l,int r,int rt)
{
	if(seg[rt].l==l&&seg[rt].r==r)
	{
		sum[rt]=c*(r-l+1);//	//把(l,r)中的元素都替换成c
		lazy[rt]=c;//先不更新子节点 lazy标记
		return; 
	}

	//要分解区间时 把以前没更新的先更新 
	Pushdown(rt,seg[rt].r-seg[rt].l+1);
	
	int m=seg[rt].mid();
	//要更新的在节点左边 
	if(r<=m)
	{
		update(c,l,r,rt<<1); 
	}
	else if(l>m)
	{
		update(c,l,r,rt<<1|1);
	}
	else//要更新的横跨中点 
	{
		update(c,l,m,rt<<1);
		update(c,m+1,r,rt<<1|1);
	}
	pushup(rt);
}
long long query(int l,int r,int rt)  
{  
    if(l==seg[rt].l&&r==seg[rt].r)//完全包含   
    {  
        return sum[rt];  
    }  
    //要用到子区间  
    Pushdown(rt,seg[rt].r-seg[rt].l+1);  
    int m=seg[rt].mid();  
    long long res=0;  
    //分解区间   
    if(r<=m)   
    {  
        res+=query(l,r,rt<<1);  
     }   
     else if(l>m)//所查询区间在右半段   
     {  
        res+=query(l,r,rt<<1|1);  
     }  
     else//l<m<r<tree[rt].r  
     {  
        res+=query(l,m,rt<<1);  
        res+=query(m+1,r,rt<<1|1);  
     }  
     return res;  
}  



int main()
{
	int t;
	cin>>t;
	for(int i=1;i<=t;i++)
	{
		int n;
		cin>>n;
		build(1,n,1);
		
		int q;
		cin>>q;
		while(q--)
		{
			int l,r,val;
		
			scanf("%d%d%d",&l,&r,&val);
			update(val,l,r,1);
		}
		printf("Case %d: The total value of the hook is %lld.\n",i,query(1,n,1));	
	}
	return 0;
}