给定长度为 n 的序列 {a},m 此操作,每一次操作将一段区间 [l,r] 赋值为 x,求每一次操作后的结果。
用线段树做了一下,结果输出全是 0,lazy_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("");
}
}