hdu1698 Just a Hook(线段树/区间赋值)

题目

1-n个数,初始全为1,

m个操作,第i次操作把[li,ri]赋成xi,1<=xi<=3

求m次操作后这n个数的和

1<=n<=1e5,0<=m<=1e5

思路来源

https://blog.csdn.net/zhouzi2018/article/details/81089816

题解

区间赋值

近乎裸的线段树板子,但之前没写过区间赋值,就总结一下吧

避免自己以后不会的时候,没有地方回看

cov[]代表这一段区间的值应该是几,

然后完整区间就直接打标记和打区间和的值

下放标记值时,将左子和右子的标记直接打成该区间赋的值,

并将左子区间和右子区间的和更新

代码

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
const int maxn=5e5+10;
int T,n,q;
ll dat[maxn],cov[maxn];
void pushup(int p)
{
	dat[p]=dat[p<<1]+dat[p<<1|1]; 
} 
void build(int p,int l,int r)
{
	cov[p]=0;
	if(l==r)
	{
		dat[p]=1;
		return;
	}
	int mid=(l+r)>>1;
	build(p<<1,l,mid);
	build(p<<1|1,mid+1,r);
	pushup(p);
}
void pushdown(int p,int l,int r)
{
	if(cov[p])
	{
		int mid=(l+r)>>1;
		dat[p<<1]=(mid-l+1)*cov[p];
		dat[p<<1|1]=(r-mid)*cov[p];
		cov[p<<1]=cov[p];
		cov[p<<1|1]=cov[p];
		cov[p]=0;
	}
}
void update(int p,int l,int r,int ql,int qr,ll x)//区间赋值 
{
	if(ql<=l&&r<=qr)
	{
		dat[p]=(r-l+1)*x;
		cov[p]=x;
		return;
	}
	pushdown(p,l,r);
	int mid=(l+r)>>1;
	if(ql<=mid)update(p<<1,l,mid,ql,qr,x);
	if(qr>=mid+1)update(p<<1|1,mid+1,r,ql,qr,x);
	pushup(p);
}
ll ask(int p,int l,int r,int ql,int qr)
{
	if(ql<=l&&r<=qr)return dat[p];
	pushdown(p,l,r);
	ll sum=0;
	int mid=(l+r)>>1;
	if(ql<=mid)sum+=ask(p<<1,l,mid,ql,qr);
	if(qr>=mid+1)sum+=ask(p<<1|1,mid+1,r,ql,qr);
	return sum;
}
int main()
{
	scanf("%d",&T);
	for(int k=1;k<=T;++k)
	{
	scanf("%d",&n);
	build(1,1,n);
	scanf("%d",&q); 
	while(q--)
	{
		int l,r,x;
		scanf("%d%d%d",&l,&r,&x);
		update(1,1,n,l,r,x);
	}
	printf("Case %d: The total value of the hook is %lld.\n",k,ask(1,1,n,1,n));
    }
	return 0;
}

猜你喜欢

转载自blog.csdn.net/Code92007/article/details/88746032