hihoCoder #1078 : 线段树的区间修改

题目大意及分析:

线段树成段更新裸题。

代码如下:

# include<iostream>
# include<cstdio>
# include<cstring>
# include<algorithm>
using namespace std; const int N=100000; int n,m;
int a[N+5];
int tr[N*4+5];
int lazy[N*4+5]; void pushDown(int rt,int l,int r)
{
if(lazy[rt]!=-1){
int mid=l+(r-l)/2;
tr[rt<<1]=lazy[rt]*(mid-l+1);
tr[rt<<1|1]=lazy[rt]*(r-mid);
lazy[rt<<1]=lazy[rt<<1|1]=lazy[rt];
lazy[rt]=-1;
}
} void makeTree(int rt,int l,int r)
{
if(l==r){
scanf("%d",&tr[rt]);
}else{
int mid=l+(r-l)/2;
makeTree(rt<<1,l,mid);
makeTree(rt<<1|1,mid+1,r);
tr[rt]=tr[rt<<1]+tr[rt<<1|1];
}
} void update(int rt,int l,int r,int L,int R,int x)
{
if(L<=l&&r<=R){
lazy[rt]=x;
tr[rt]=x*(r-l+1);
}else{
pushDown(rt,l,r);
int mid=l+(r-l)/2;
if(L<=mid)
update(rt<<1,l,mid,L,R,x);
if(R>mid)
update(rt<<1|1,mid+1,r,L,R,x);
tr[rt]=tr[rt<<1]+tr[rt<<1|1];
}
} int query(int rt,int l,int r,int L,int R)
{
if(L<=l&&r<=R)
return tr[rt];
pushDown(rt,l,r);
int res=0;
int mid=l+(r-l)/2;
if(L<=mid)
res+=query(rt<<1,l,mid,L,R);
if(R>mid)
res+=query(rt<<1|1,mid+1,r,L,R);
return res;
} int main()
{
while(~scanf("%d",&n))
{
memset(tr,0,sizeof(tr));
memset(lazy,-1,sizeof(lazy));
makeTree(1,1,n);
scanf("%d",&m);
int flag,a,b,c;
while(m--)
{
scanf("%d",&flag);
if(flag){
scanf("%d%d%d",&a,&b,&c);
update(1,1,n,a,b,c);
}else{
scanf("%d%d",&a,&b);
printf("%d\n",query(1,1,n,a,b));
}
}
}
return 0;
}

  

上一篇:C++陷阱之慎用string类


下一篇:C# 跳转新的标签页