poj3468

最新推荐文章于 2018-02-28 16:06:38 发布

原创最新推荐文章于 2018-02-28 16:06:38 发布 · 922 阅读

0 ·

CC 4.0 BY-SA版权

数据结构专栏收录该内容

85 篇文章

订阅专栏

本文详细解析了线段树lazy操作的题解过程，包括构建、更新和查询操作，适用于初学者深入理解线段树的应用。

/*
分析：
线段树lazy操作的果题，不多嗦了。。

2013-04-09
*/

#include"iostream"
#include"cstdio"
#include"cstring"
using namespace std;
const int N=100011;

struct Segtree
{
	int l,r,mid;
	int flag;
	__int64 count,sum;
}T[N<<2];
int ttt[N];

void push_down(int k)
{
	T[2*k].flag=T[2*k+1].flag=1;
	T[2*k].count+=T[k].count;
	T[2*k+1].count+=T[k].count;
	T[2*k].sum+=(T[2*k].r-T[2*k].l+1)*T[k].count;
	T[2*k+1].sum+=(T[2*k+1].r-T[2*k+1].l+1)*T[k].count;
	T[k].count=T[k].flag=0;
}
void build(int l,int r,int k)
{
	T[k].l=l;
	T[k].r=r;
	T[k].mid=(l+r)>>1;
	T[k].count=T[k].flag=0;
	if(l==r)	{T[k].sum=ttt[l];return ;}
	build(l,T[k].mid,2*k);
	build(T[k].mid+1,r,2*k+1);
	T[k].sum=T[2*k].sum+T[2*k+1].sum;
}
void update(int l,int r,int dir,int k)
{
	if(T[k].l==l && T[k].r==r)
	{
		T[k].count+=dir;
		T[k].sum+=(r-l+1)*dir;
		T[k].flag=1;
		return ;
	}
	if(T[k].flag)	push_down(k);
	if(r<=T[k].mid)		update(l,r,dir,2*k);
	else if(l>T[k].mid)	update(l,r,dir,2*k+1);
	else
	{
		update(l,T[k].mid,dir,2*k);
		update(T[k].mid+1,r,dir,2*k+1);
	}
	T[k].sum=T[2*k].sum+T[2*k+1].sum;
}
__int64 query(int l,int r,int k)
{
	__int64 ans=0;
	if(T[k].l==l && T[k].r==r)	return T[k].sum;
	if(T[k].flag)	push_down(k);
	if(r<=T[k].mid)		ans=query(l,r,2*k);
	else if(l>T[k].mid)	ans=query(l,r,2*k+1);
	else
	{
		ans=query(l,T[k].mid,2*k);
		ans+=query(T[k].mid+1,r,2*k+1);
	}
	return ans;
}

int main()
{
	int n,m;
	int i;
	char str[11];
	int a,b,c;
	while(scanf("%d%d",&n,&m)!=-1)
	{
		for(i=1;i<=n;i++)	scanf("%d",&ttt[i]);
		build(1,n,1);
		while(m--)
		{
			scanf("%s",str);
			if(str[0]=='Q')
			{
				scanf("%d%d",&a,&b);
				printf("%I64d\n",query(a,b,1));
			}
			else
			{
				scanf("%d%d%d",&a,&b,&c);
				update(a,b,c,1);
			}
		}
	}
	return 0;
}