【模板】 线段树

解决问题

区间修改与区间查询问题，期望将单次查询或修改的时间复杂度 降至 O(log n)

整体思路

Part1: 根据二叉树性质建树 L_son=pos2,R_son=pos2+1
Part2： 区间修改： 从(1,n）中找区间(l,r），修改其add值，add值保留但不下放
Part3: 区间查询， 从(1,n）中找区间(l,r），计算其sum值
注：当修改或查询时，如果遇到未下放的add，则实时下放

貌似并没有不好理解的地方
当时学不会可能是因为长代码细节太多容易出错
【没放带区间乘法的…]

Code

自写read()

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
//线段树 区间加与区间求和 
inline int read(){
	char ch=getchar();int sum=0;
	while(ch<'0'||ch>'9') ch=getchar();
	while(ch>='0'&&ch<='9')
		sum*=10,sum+=ch-'0',ch=getchar();
	return sum;
}

建树

const int N=100005;int n,m;
long long s[4*N+10],a[4*N+10];
inline void up(int pos){s[pos]=s[2*pos+1]+s[2*pos];}//求和
inline void build(int pos,int l,int r){//建树
	if(l==r) {s[pos]=read();return;}
	int mid=(l+r)/2;
	build(2*pos,l,mid);
	build(2*pos+1,mid+1,r);
	up(pos);	return ;
}

lazy下放

inline void down(int pos,int l,int r){//下放
	if(!a[pos]) return;
	a[2*pos]+=a[pos];a[2*pos+1]+=a[pos];
	int mid=(l+r)/2;
	s[2*pos]+=(mid-l+1)*a[pos];
	s[2*pos+1]+=(r-mid)*a[pos];
	a[pos]=0; return ;
}

区间修改

void push_up(int pos,int l,int r,int va,int y,int x){//区间修改
	if(x<=l&&r<=y){
		s[pos]+=(r-l+1)*va;
		a[pos]+=va;	return;
	}
	int mid=(l+r)/2; down(pos,l,r);
	if(x<=mid) push_up(2*pos,l,mid,va,y,x);
	if(mid+1<=y) push_up(2*pos+1,mid+1,r,va,y,x);
	up(pos); 	return;
}

区间查询

long long query(int pos,int l,int r,int y,int x){//区间查询
	if(x<=l&&r<=y) return s[pos];
	down(pos,l,r);int mid=(l+r)/2;
	long long ret=0;
	if(x<=mid) ret+=query(2*pos,l,mid,y,x);
	if(mid+1<=y) ret+=query(2*pos+1,mid+1,r,y,x);
	return ret;
}

主函数

int main()
{
	n=read(),m=read();build(1,1,n);
	for(int i=1;i<=m;i++)
	if(1==read())
		push_up(1,1,n,read(),read(),read());
	else cout<<query(1,1,n,read(),read())<<endl;
	return 0;
}