这次是把某个区间全部更新为一个值,而不是增加或减少。最后询问一下区间总和。
思路还是差不多的。在更新时,当我们将一个节点所维护的区间更新后,可以用这个区间的长度*新的值,即
tree[x].sum = (tree[x].r-tree[x].l+1)*tree[x].data。
在查询或更新区间时,可以将tree[x].data作为懒惰标记继续往下传,最后完成push_up操作:
tree[x].sum = tree[x<<1].sum+tree[x<<1|1].sum;
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
const int maxn = 100000+5;
using namespace std;
int t,n,q,x,y,z;
struct node{
int l,r;
int sum,data;
void update(int x)
{
data = x;
}
}tree[maxn*4];
void getsum(int x)
{
tree[x].sum = (tree[x].r-tree[x].l+1)*tree[x].data;
}
void push_down(int x)
{
int lazyval = tree[x].data;
if(lazyval)
{
tree[x<<1].update(lazyval);
getsum(x<<1);
tree[x<<1|1].update(lazyval);
getsum(x<<1|1);
tree[x].data = 0;
}
}
void build(int x,int l,int r)
{
tree[x].l = l,tree[x].r = r;
if(l==r){
tree[x].sum = 1;
tree[x].data = 1;
}
else{
int mid = (l+r)/2;
build(x<<1,l,mid);
build(x<<1|1,mid+1,r);
tree[x].data = 0;
tree[x].sum = tree[x<<1].sum+tree[x<<1|1].sum;
}
}
void update(int x,int l,int r,int z)
{
int L=tree[x].l,R=tree[x].r;
if(l<=L&&r>=R)
{
tree[x].data = z;
getsum(x);
}
else{
push_down(x);
int mid = (L+R)/2;
if(l<=mid)
update(x<<1,l,r,z);
if(r>mid)
update(x<<1|1,l,r,z);
tree[x].sum = tree[x<<1].sum+tree[x<<1|1].sum;
}
}
long long query(int x,int l,int r)
{
int L=tree[x].l,R=tree[x].r;
long long ans = 0;
if(l<=L&&r>=R)
ans += tree[x].sum;
else{
push_down(x);
int mid = (L+R)/2;
if(l<=mid)
ans+=query(x<<1,l,r);
if(r>mid)
ans+=query(x<<1|1,l,r);
getsum(x);
//cout<<ans<<endl;
}
return ans;
}
int main()
{
freopen("d://test.txt","r",stdin);
scanf("%d", &t);
for(int j=1;j<=t;j++)
{
scanf("%d", &n);
build(1,1,n);
scanf("%d",&q);
for(int i=0;i<q;i++)
{
scanf("%d%d%d",&x,&y,&z);
update(1,x,y,z);
}
// for(int i=1;i<=n;i++)
// cout<<tree[i].sum<<endl;
printf("Case %d: The total value of the hook is %d.\n",j,tree[1].sum);
}
return 0;
}