【算法学习笔记】线段树

【算法学习笔记】线段树

线段树是用来维护区间信息的数据结构，可以在O(logN)内进行区间修改，区间查询等操作。
需要注意的点比较多，结合代码进行说明。

#include<cstdio>

using namespace std;

//树节点
//线段树的每一个节点代表一个区间
//根节点覆盖整个区间，其左儿子覆盖该区间的左半部分，右儿子覆盖右半部分
//以此类推，叶子节点只覆盖一个点
struct node{
	int l, r;//覆盖区间范围
	long long val, lazy;
}tree[400005];

int n, m, a[100005], temp;
long long ans;

//建树操作
void build(int k, int tl, int tr)
{
	tree[k].l=tl, tree[k].r=tr;
	if(tl==tr)//是叶子节点
	{
		tree[k].val=a[tl];
		return;
	}
	int mid=(tl+tr)>>1;
	//分别递归左儿子和右儿子建树
	build(2*k, tl, mid);
	build(2*k+1, mid+1, tr);
	//线段树维护的是区间的信息
	//父亲节点的区间信息可由其两个儿子节点的信息相加得到
	tree[k].val=tree[2*k].val+tree[2*k+1].val;
}

//插播对懒标记的解释
//如果我们在进行区间修改时仍然把遍历每个节点，那时间复杂度其实和普通的遍历没有区别
//而线段树的结构本身就是基于区间的，所以在区间求值时我们有时并不需要一路修改到叶子节点
//为了利用好线段树这一特点，我们引入懒标记
//当我们修改区间时，只将能恰好覆盖这一区间所需的所有节点的值改变。
//对于他们的子节点，我们不做修改，而是把父亲节点的懒标记加上修改的数值。
//之后查询的区间如果需要访问他的子节点，就把懒标记下放给子节点，对其进行修改
//较为形象的理解是把懒标记看作父亲节点对儿子节点的“欠条”。
//当前的查询不需要访问儿子节点，就不更新他，但会做个标记，以后需要访问他的时候再给他
//一个有懒标记的节点，其自身维护的区间已经更新完毕，但儿子节点还未更新

//懒标记下放操作
void down(int k)
{
	tree[2*k].lazy+=tree[k].lazy;
	tree[2*k+1].lazy+=tree[k].lazy;
	tree[2*k].val+=tree[k].lazy*(tree[2*k].r-tree[2*k].l+1);
	tree[2*k+1].val+=tree[k].lazy*(tree[2*k+1].r-tree[2*k+1].l+1);
	//懒标记下放后，要记得把父节点的懒标记清零，避免重复计算
	tree[k].lazy=0;
}

//区间修改
void change(int k, int x, int tl, int tr)
{
	//再次强调，线段树的节点是区间
	//所以要找到完全在所给区间内的节点再进行修改
	
	//在所给区间内，修改区间和，更新懒标记
	if(tree[k].l>=tl&&tree[k].r<=tr)
	{
		tree[k].val+=(tree[k].r-tree[k].l+1)*x;
		tree[k].lazy+=x;
		return;
	}
	//懒标记表示该节点的值已处理好，为真就直接向儿子节点传递
	if(tree[k].lazy)
		down(k);
	//如果当前区间并未完全包含于所给区间，我们就需要遍历他的儿子节点，找到符合要求的节点
	int mid=(tree[k].l+tree[k].r)>>1;
	//区间中点位于左端点右侧，则左区间部分或全部位于两端点之间
	if(tl<=mid)
		change(2*k, x, tl, tr);
	//区间中点位于右端点左侧，则有区间部分或全部位于两端点之间
	if(tr>mid)
		change(2*k+1, x, tl, tr);
	//向下遍历儿子节点后，记得更新父亲节点
	tree[k].val=tree[2*k].val+tree[2*k+1].val;
}

//区间查询
//思想和区间修改类似
void ask(int k, int tl, int tr)
{
	//当前区间位左右端点之间，更新ans
	if(tree[k].l>=tl&&tree[k].r<=tr)
	{
		ans+=tree[k].val;
		return;
	}
	if(tree[k].lazy)
		down(k);
	int mid=(tree[k].l+tree[k].r)>>1;
	if(tl<=mid)
		ask(2*k, tl, tr);
	if(tr>mid)
		ask(2*k+1, tl, tr);
}

int main(void)
{
	scanf("%d%d", &n, &m);
	for(int i=1; i<=n; i++)
		scanf("%d", &a[i]);
	build(1, 1, n);
	for(int i=1; i<=m; i++)
	{
		int x, y, k;
		ans=0;
		scanf("%d", &temp);
		//区间修改
		if(temp==1)
		{	
			scanf("%d%d%d", &x, &y, &k);
			change(1, k, x, y);
		}
		//区间查询
		else if(temp==2)
		{
			scanf("%d%d", &x, &y);
			ask(1, x, y);
			printf("%lld\n", ans);
		}
	}
	return 0;
}