hdu1968 区间更新+区间求和(线段树)

    这次是把某个区间全部更新为一个值,而不是增加或减少。最后询问一下区间总和。

    思路还是差不多的。在更新时,当我们将一个节点所维护的区间更新后,可以用这个区间的长度*新的值,即

                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
#include
#include
#include
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<

你可能感兴趣的:(线段树)