poj3468线段树区间求和 +区间更新(模板)
#include <iostream>
#include <algorithm>
#include <string.h>
#include <stdio.h>
#define L o<<1//*2
#define R o<<1|1//*2+1
#define ll long long
using namespace std;
struct node
{
int l,r;//区间
long long sum;//区间和
long long lazy;//延迟值
int length;//区间长度
}Tree[100000*4];//一般取需求值的4倍
void pushon(int o)
{
Tree[o].sum=Tree[L].sum+Tree[R].sum;//归并最大值
}
void build(int o ,int l ,int r)
{
Tree[o].l=l;
Tree[o].r=r;
Tree[o].lazy=0;
Tree[o].length=r-l+1;
if(l==r)
{
scanf("%lld",&Tree[o].sum);
return;//直接返回
}
int mid=(l+r)/2;//只有创建树的时候是(l+r)/2
build(L,l,mid);//和归并类似,不过函数前多一个L,R
build(R,mid+1,r);
pushon(o);//归并区间的总和
}
void pushdown(int o)//更新区间值
{
if(Tree[o].lazy)
{
Tree[L].sum=Tree[L].sum+Tree[L].length*Tree[o].lazy;//原来的+区间长度*增加的值
Tree[R].sum=Tree[R].sum+Tree[R].length*Tree[o].lazy;
Tree[L].lazy+=Tree[o].lazy;
Tree[R].lazy+=Tree[o].lazy;
Tree[o].lazy=0;
}
}
void update(int o,int l ,int r,int lazy)
{
if(Tree[o].l==l&&Tree[o].r==r)
{
Tree[o].sum+=lazy*Tree[o].length;//这里是函数的lazy
Tree[o].lazy+=lazy;
return;
}
pushdown(o);//更新区间延迟值
int mid=(Tree[o].l+Tree[o].r)/2;//这里是Tree[o].l .r注意!
if(mid>=r)
{
update(L,l,r,lazy);
}
else if(l>mid)
{
update(R,l,r,lazy);
}
else
{
update(L,l,mid,lazy);
update(R,mid+1,r,lazy);
}
pushon(o);//更新总体区间sum值
}
long long Query(int o,int l,int r)
{
if(Tree[o].l==l&&Tree[o].r==r)
{
return Tree[o].sum;//恰好等于区间长度
}
pushdown(o);//更新延迟值,防止之前没更新
int mid=(Tree[o].l+Tree[o].r)/2;//这里是Tree[o].l .r注意!
if(mid>=r)//相当与最小值比 最右边区间r还大
{
return Query(L,l,r);//l-r
}
else if(l>mid) //相当与最大值比 最左边区间l还小
{
return Query(R,l,r);//l-r
}
else
{
return (Query(L,l,mid)+Query(R,mid+1,r));
}
}
int main()
{
int n,m,x,y,z;
scanf("%d%d",&n,&m);
build