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

解题思路：基础的线段树区间修改

我按照书上敲的代码不知道为什么WA。。。

#include
#include
#include
using namespace std;

const int maxn = 1e5;
int n,q,l,r,_sum;
int setv[maxn<<2],sum[maxn<<2];

void maintain(int o,int L,int R)
{
	int lc = o*2, rc = o*2+1;
	sum[o] = 0;
	if(R > L){
		sum[o] = sum[lc] + sum[rc];
	}
	else sum[o] = setv[o];
}

void pushdown(int o)
{
	int lc = o*2, rc = o*2+1;
	if(setv[o] >= 0){
		setv[lc] = setv[rc] = setv[o];
		setv[o] = -1;
	}
}

void update(int o,int L,int R,int v)
{
	int lc = o*2, rc = 2*o+1;
	if(l <= L && R <= r){
		setv[o] = v;
	}
	else{
		pushdown(o);
		int M = (L + R) >> 1;
		if(l <= M) 
			update(lc,L,M,v);
		else maintain(lc,L,M);
		if(r > M)
			update(rc,M+1,R,v);
		else maintain(rc,M+1,R);
	}
	maintain(o,L,R);
}

void query(int o,int L,int R)
{
	if(setv[o] >= 0){
		_sum += setv[o] * (min(R,r) - max(L,l) + 1);
	}
	else if(l <= L && R <= r){
		_sum += sum[o];
	}
	else{
		int M = (L + R) >> 1;
		if(l <= M) query(o*2,L,M);
		if(r > M) query(o*2+1,M+1,R);
	}
}

int main()
{
	int v,f;
	scanf("%d",&n);
	memset(setv,-1,sizeof(setv));
	for(int i = 1; i <= n; i++)
	{
		scanf("%d",&v);
		l = r = i;
		update(1,1,n,v);
	}
	scanf("%d",&q);
	while(q--)
	{
		scanf("%d",&f);
		if(f == 1)
		{
			scanf("%d%d%d",&l,&r,&v);
			update(1,1,n,v);
		}
		else
		{
			scanf("%d%d",&l,&r);
			_sum = 0;
			query(1,1,n);
			printf("%d\n",_sum);
		}
	}
	return 0;
}