线段树详解

线段树这个东西,它是一棵二叉树,用途就是维护一个数列,它的思想类似于分治,分而治之,首先,这棵树的每一个节点有两个属性,l和r,表示自己管理的范围,那么这棵树的根,它的l和r自然就是1和n(n就是数列的总长度),然后,每个节点有两个儿子,分别管理 自己要管理的范围 的1/2 ,拿根节点做例子,根节点要管理1~n,那么它的左儿子就管理1~n/2,右儿子管理n/2+1~n,再记录一下自己左右儿子是谁,这棵树就建好了,比如有一个数列n=4,那么建出来的树就是这个样子的:

可以发现,建完树之后,叶子节点的管理范围只有一,那么他们管理的也就只有在数列中那个位置的数

建树代码:

void buildtree(int x,int y)
{
    len++;
    tree[len].l=x;
    tree[len].r=y;
    tree[len].zuo=tree[len].you=-1;
    tree[len].late=0;//late标记下面会讲
    if(x==y)tree[len].sum=a[x];//sum表示在数列中它管理范围内的数的和
    else
    {
        int now=len;
        int mid=(x+y)/2;//将自己的管理区间分给两个儿子(分)
        tree[now].zuo=len+1;buildtree(x,mid);
        tree[now].you=len+1;buildtree(mid+1,y);
        tree[now].sum=tree[tree[now].zuo].sum+tree[tree[now].you].sum;//由儿子的值得到自己的值(合并)
    }
}

 

修改和查询(这里以求区间和为例子):

 

修改:修改时,我们只需要看看要修改的区间在当前的节点的左儿子上还是在右儿子上,如果只在某一个儿子上,那么直接去修改那个儿子就好了,但如果在两个儿子上都有,那么就把左儿子管理的那些丢给左儿子,右儿子管理的那些丢给右儿子就好了。

比如当前我们需要让1~2区间+x,那么首先从根节点开始,发现1~2都是在左儿子的管理范围里,于是就去到根的左儿子那里,然后发现左儿子的管理范围刚刚好就是要修改的范围,那么这个时候,我们就不需要继续往下了,只需要将这个点的late+x即可,等我们下一次再访问到这个节点时,我们再把它的late值扔给儿子就好了,那么这样子就可以把修改的时间复杂度从O(nlogn)优化成O(玄学)了。。。

代码如下:

void change(int now,int x,int y,ll z)//now表示现在到了线段树的哪个节点,x和y表示修改范围
{
    if(tree[now].l==x&&tree[now].r==y)//管理范围刚好等于修改范围
    {
        tree[now].late+=z;
        give(now);//分配下去
        return;
    }//假如这个点的管理范围比修改范围要大
    give(now);
    int zuo=tree[now].zuo,you=tree[now].you;
    int mid=(tree[now].l+tree[now].r)/2;
    if(y<=mid)change(zuo,x,y,z);//如果在左儿子那里
    else if(x>=mid+1)change(you,x,y,z);//如果在右儿子那里
    else//加入两边都有,就各自修改
    {
        change(zuo,x,mid,z);
        change(you,mid+1,y,z);
    }
    tree[now].sum+=z*(y-x+1);//!!!自己有(y-x+1)个儿子加了z,那么自己的sum值就要加上(y-x+1)* z
} 

查询:

查询和修改差不多,直接看代码吧:

ll print(int now,int x,int y)//x,y表示要求x到y的区间和
{
    give(now);//分配下去,一定要先分配,想想就知道为什么了
    if(tree[now].l==x&&tree[now].r==y)return tree[now].sum;//假如要求的区间刚好就是这个节点管理的区间,那么只需要返回它的sum就好了
    int zuo=tree[now].zuo,you=tree[now].you;
    int mid=(tree[now].l+tree[now].r)/2;//否则就继续往下找
    if(y<=mid)return print(zuo,x,y);
    else if(x>=mid+1)return print(you,x,y);
    else return print(zuo,x,mid)+print(you,mid+1,y);
}

最后补上give函数:

void give(int x)
{
	int zuo=tree[x].zuo,you=tree[x].you;
	tree[x].sum+=(tree[x].r-tree[x].l+1)*tree[x].late;//自己的管理范围内的数全部都加了tree[x].late,那么自己的值自然就要加上 (管理范围*tree[x].late)
	if(zuo!=-1)//假如有儿子,就把自己的late分配下去
	{
		tree[zuo].late+=tree[x].late;
		tree[you].late+=tree[x].late;
	}
	tree[x].late=0;//清空late标记
}

如果是改点的话,late就是废的。

线段树因为十分灵活,所以,它可以和很多结构一起用,也可以有很多千奇百怪的用法。例如,树链剖分,它的用处也是很广泛。还可以有更厉害的用法,主席树

要注意,线段树的用途不仅是处理数列,它最重要的,是对连续区间的管理——对区间的整体操作,以及对区间信息的收集和处理。

个人建议线段树用指针版的打法,既短,又好看。

完整改段求段代码:

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long

int n,m,a[100010];
struct node{
	int l,r,mid;ll sum,lazy;node *zuo,*you;
	node(int x,int y):l(x),r(y),mid(x+y>>1),lazy(0){
		if(l<r){
			zuo=new node(l,mid);
			you=new node(mid+1,r);
			sum=zuo->sum+you->sum;
		}else zuo=you=NULL,sum=a[x];
	}
	void pushdown(){
		if(zuo)zuo->lazy+=lazy,zuo->sum+=(mid-l+1)*lazy,
		you->lazy+=lazy,you->sum+=(r-mid)*lazy;
		lazy=0;
	}
	void change(int x,int y,ll z){
		pushdown();
		if(l==x&&r==y){lazy+=z;sum+=z*(y-x+1);pushdown();return;}
		if(y<=mid)zuo->change(x,y,z);
		else if(x>=mid+1)you->change(x,y,z);
		else zuo->change(x,mid,z),you->change(mid+1,y,z);
		sum=zuo->sum+you->sum;
	}
	ll getsum(int x,int y){
		pushdown();
		if(l==x&&r==y)return sum;
		if(y<=mid)return zuo->getsum(x,y);
		else if(x>=mid+1)return you->getsum(x,y);
		else return zuo->getsum(x,mid)+you->getsum(mid+1,y);
	}
}*root=NULL;

int main()
{
	scanf("%d %d",&n,&m);
	for(int i=1;i<=n;i++)scanf("%d",&a[i]);
	root=new node(1,n);
	int op,x,y;ll z;
	for(int i=1;i<=m;i++){
		scanf("%d %d %d",&op,&x,&y);
		if(op==1)scanf("%lld",&z),root->change(x,y,z);
		else printf("%lld\n",root->getsum(x,y));
	}
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值