关于线段树
  • 板块灌水区
  • 楼主parallet
  • 当前回复21
  • 已保存回复21
  • 发布时间2020/8/27 09:56
  • 上次更新2023/11/6 19:12:56
查看原帖
关于线段树
365714
parallet楼主2020/8/27 09:56

给定长度为 nn 的序列 {a}\{a\}mm 此操作,每一次操作将一段区间 [l,r][l,r] 赋值为 xx,求每一次操作后的结果。

用线段树做了一下,结果输出全是 00lazy_taglazy\_tag 貌似写假了,求差错!

#include<cstdio>
using namespace std;
const int MAXN=4000010;
int n,m,a[MAXN],tree[MAXN],tag[MAXN];
inline int lc(int p){return p<<1;}
inline int rc(int p){return p<<1|1;}
void push_up(int u)
{
	tree[u]=tree[lc(u)]+tree[rc(u)];
}
void lazy_tag(int u,int l,int r,int k)
{
	tag[u]=k;
	tree[u]=k*(r-l+1);
}
void push_down(int u,int l,int r)
{
	int mid=(l+r)>>1;
	lazy_tag(lc(u),l,mid,tag[u]);
	lazy_tag(rc(u),mid+1,r,tag[u]);
	tag[u]=0;
}
void build(int u,int l,int r)
{
	if(l==r)
	{
		tree[u]=a[l];
		return;
	}
	int mid=(l+r)>>1;
	build(lc(u),l,mid);
	build(rc(u),mid+1,r);
	push_up(u);
}
void update(int u,int l,int r,int ln,int rn,int k)
{
	if(ln<=l&&r<=rn)
	{
		lazy_tag(u,l,r,k);
		return;
	}
	push_down(u,l,r);
	int mid=(l+r)>>1;
	if(ln<=mid) update(lc(u),l,mid,ln,rn,k);
	if(rn>mid) update(rc(u),mid+1,r,ln,rn,k);
	push_up(u);
}
int query(int u,int l,int r,int ln,int rn)
{
	int t=0;
	if(ln<=l&&r<=rn) return tree[u];
	push_down(u,l,r);
	int mid=(l+r)>>1;
	if(ln<=mid) t+=query(lc(u),l,mid,ln,rn);
	if(rn>mid) t+=query(rc(u),mid+1,r,ln,rn);
	return t;
}
int main()
{
	scanf("%d",&n);
    for(register int i=1;i<=n;i++) scanf("%d",&a[i]);
    build(1,1,n);
    scanf("%d",&m);
    while(m--)
    {
    	int l,r,x;
    	scanf("%d%d%d",&l,&r,&x);
    	update(1,1,n,l,r,x);
    	for(register int i=1;i<=n;i++) printf("%d ",query(1,1,n,i,i));
    	puts("");
	}
}

2020/8/27 09:56
加载中...