线段树模板

自用线段树模板,适用于基本线段树和成段更新...

/* *************************************************************** *\
|
| 基础线段树模板
| push_up(),build(),query()为基本线段树函数
| push_down()和update()函数用于成段更新
|
\* *************************************************************** */

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <string.h>

using namespace std;

#define maxn 30000				//叶子节点数
#define Mid int mid=(left+right)>>1	//定义mid
#define lson rt*2,left,mid
#define rson rt*2+1,mid+1,right

int n,m;					//n叶子节点数,m询问数
int node[maxn<<2];		//数组开maxn的四倍
int lazy[maxn<<2];			//懒惰标记

void push_up(int rt)			//用于更新节点信息,修改时需要同时修改query()中返回值的处理机制,使二者统一(成段更新时还需修改push_down())
{
node[rt]=node[rt*2]+node[rt*2+1];	//这部分由线段树的作用(求和,求乘积,求最值......)决定
}

void push_down(int rt,int left ,int right)		//懒惰标记下移一层
{
Mid;
lazy[rt*2]+=lazy[rt];
node[rt*2]=node[rt*2]+lazy[rt]*(mid-left+1);
lazy[rt*2+1]+=lazy[rt];
node[rt*2+1]=node[rt*2+1]+lazy[rt]*(right-mid);
lazy[rt]=0;
push_up(rt);
}

void build (int rt ,int left , int right )		//建树
{
if (left==right)
{
scanf("%d",&node[rt]);
return ;
}
Mid;
build (lson);
build(rson);
push_up(rt);
}

void update(int rt ,int left ,int right ,int l,int r,int c)		//用于更新区间[left,right]的节点的值
{
if (l<=left&&right<=r)
{
lazy[rt]+=c;
node[rt]+=c*(right-left+1);
return ;
}
push_down(rt,left,right);
Mid;
if (l<=mid)
update(lson,l,r,c);
if (r>mid)
update(rson,l,r,c);
push_up(rt);
}

int query (int rt ,int left ,int right ,int l,int r)
{
if (l<=left&&right<=r)
{
return node[rt];
}
//push_down(rt,left,right);			//成段更新时使用
Mid;
int ret=0;
if (l<=mid)
ret+=query(lson,l,r);
if (r>mid)
ret+=query(rson,l,r);
return ret;
}

int main()
{
int l,r,flg=0;
while (scanf("%d%d",&n,&m)!=EOF)
{
memset(node,0,sizeof (node));
memset(lazy,0,sizeof (lazy));
build(1,1,n);
for (int i=0;i<m;i++)
{
scanf("%d%d",&l,&r);
printf("%d\n",query(1,1,n,l,r) );
}
}
return 0;
}