线段树//洛谷P3372

第一行国际惯例咕咕咕。

第二行感谢csb师哥。

第三行——我终于明白,我长这么大一直活得特开心的最大的原因就是擅长知难而退、懂得敢于放弃。(此处当然是反讽

线段树:

1.定义:一种二叉搜索树(即最多只有左右节点),利用二分的思想每个节点储存的是一个区间值(其实就是一条线段啦)。叶子节点存每个点(区间左右端点相等),其余内部节点存区间[l,mid]和[mid+1,r](相对其父节点的左右节点)。

2.原理:将n个点分为4*n个特定的子区间,通过对子区间的修改,进行l到r之间的修改统计。

3.存储:用数组储存(这个比较方便简单)(你要愿意用指针也不是不行但我指针学的不好我不用)。大小应当为4*n,n为节点个数(实际上只需要找到大于n的最小2的幂,乘2即可。但只要不mle,n*4就行,省事儿)。每个数组存的是从l到r的连续区间和。只要l<r就持续分解成两个区间,直到叶子节点上为一个点。

4.具体操作包括构造、修改、查询。采用递归的方法构造,

5.引进懒惰数组(当区间修改时。

由于我们要做的操作是区间加一个数,所以在区间进行修改时为该区间打上一个标记,就不必再修改他的儿子所维护区间,等到要使用该节点的儿子节点维护的值时,再将懒标记下放即可,可以节省很多时间,对于每次区间修改和查询,将懒标记下传,可以节省很多时间。

在我的理解中,就是表示区间元素个数(反正是按区间进行修改。所以当前区间要+/-的数,为该区间的长度*更改的数值。记录下父节点的懒惰标记后,即可跳出进行下次修改,因为左子树右子树的长度/端点都是规定好的,可以利用父节点的懒惰数组进行子节点的修改,节省时间.

6.时间复杂度为log2n。

递归爆栈了我哭辽


先附上各种模板

#define ll long long
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
ll tree[maxn*4+5] = {0};//线段树数组
ll mark[maxn*4+5] = {0};//懒惰标记
ll a[maxn+5];//初值

建树(递归):

void build(ll l, ll r, ll rt)
{//rt是当前节点
    if(l == r)
    {
        tree[rt] = a[l];
        return;
    }//为叶子节点,直接赋值返回
    ll m = (l+r)>>1;
    build(lson);//左子树
    build(rson);//右子树
    pushup(rt);//建好树后向上更新线段树节点代表的区间和
}

求和:

void pushup(ll rt)
{
    tree[rt] = tree[rt<<1]+tree[rt<<1|1];
}//当前节点的值为左右子节点的值的和

更新懒惰标记:

//由于我们要做的操作是某个区间全部加一个数
//所以我们不妨在区间进行修改时为该区间打上一个标记
//就不必再修改他的儿子所维护区间
//等到要使用该节点的儿子节点维护的值时
//再将懒标记下放即可,可以节省很多时间

void pushdown(ll l,ll m,ll r,ll rt)
{//rt为当前点,l为当前点的左端点,r为右端点,m为中间点
    if(mark[rt])//不为0,则需要更新其他点
    {
        tree[rt<<1] += mark[rt]*(m-l+1);
        tree[rt<<1|1] += mark[rt]*(r-m);
        mark[rt<<1] += mark[rt];
        mark[rt<<1|1] += mark[rt];
        mark[rt] = 0;
    }
}//tree存的是和,所以更新时要加区间内点的个数次

区间更新(某一区间内的所有点加上k):

void update(ll L,ll R,ll k,ll l,ll r,ll rt)
{//LR为目标区间,k为目标值
    if(L <= l && R >= r)
    {
        tree[rt] += (r-l+1)*k;
        mark[rt] += k;
        return;
    }//区间包含
    ll m = (l+r)>>1;
    pushdown(l,m,r,rt);//修改懒惰标记
    if(L <= m)
        update(L,R,k,lson);
    if(R > m)
        update(L,R,k,rson);
    pushup(rt);//向上更新区间和
}

单点更新(某一个点加上k):

void update(int p,int x,int l,int r,int rt)//l,r是当前节点区间,rt表示当前节点
{
    if(l == r)
    {
        tree[rt] += x;
        return;
    }//到达p点,更新叶子节点的值
    pushdown(rt,r-l+1);
    int m = (l+r)>>1;
    if(p <= m) update(p,x,lson);//根据条件判断修改左子树还是右子树
    else update(p,x,rson);
    pushup(rt);
}//表示在p点上+x

区间查询(求区间和):

ll query(ll L,ll R,ll l,ll r,ll rt)
{//LR为询问区间
    if(L <= l && R >= r)
        return tree[rt];//询问区间包含在现区间
    ll m = (l+r)>>1;
    pushdown(l,m,r,rt);//更新懒惰标记
    ll res = 0;
    if(L <= m)//左子树
        res += query(L,R,lson);
    if(R > m)//右子树
        res += query(L,R,rson);
    return res;
}

本代码以洛谷P3372为例

#include<bits/stdc++.h>
#define ll long long
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define maxn 100000

using namespace std;

ll tree[maxn*4+5] = {0};
ll mark[maxn*4+5] = {0};
ll a[maxn+5];

void build(ll l, ll r, ll rt)
{
    if(l == r)
    {
        tree[rt] = a[l];
        return;
    }
    ll m = (l+r)>>1;
    build(lson);
    build(rson);
    tree[rt] = tree[rt<<1]+tree[rt<<1|1];
}

void update(ll L,ll R,ll k,ll l,ll r,ll rt)
{
    if(L <= l && R >= r)
    {
        tree[rt] += (r-l+1)*k;
        mark[rt] += k;
        return;
    }
    ll m = (l+r)>>1;
    if(mark[rt])
    {
        tree[rt<<1] += mark[rt]*(m-l+1);
        tree[rt<<1|1] += mark[rt]*(r-m);
        mark[rt<<1] += mark[rt];
        mark[rt<<1|1] += mark[rt];
        mark[rt] = 0;
    }
    if(L <= m)
        update(L,R,k,lson);
    if(R > m)
        update(L,R,k,rson);
    tree[rt] = tree[rt<<1] + tree[rt<<1|1];
}

ll query(ll L,ll R,ll l,ll r,ll rt)
{
    if(L <= l && R >= r)
        return tree[rt];
    ll m = (l+r)>>1;
    if(mark[rt])
    {
        tree[rt<<1] += mark[rt]*(m-l+1);
        tree[rt<<1|1] += mark[rt]*(r-m);
        mark[rt<<1] += mark[rt];
        mark[rt<<1|1] += mark[rt];
        mark[rt] = 0;
    }
    ll res = 0;
    if(L <= m)
        res += query(L,R,lson);
    if(R > m)
        res += query(L,R,rson);
    return res;
}

int main()
{
    ll m,n;
    scanf("%lld%lld",&n,&m);
    for(int i = 1; i <= n; ++i)
        scanf("%lld",&a[i]);
    build(1,n,1);
    while(m--)
    {
        int op;
        scanf("%d",&op);
        if(op == 1)
        {
            ll x,y,k;
            scanf("%lld%lld%lld",&x,&y,&k);
            update(x,y,k,1,n,1);
        }
        else if(op == 2)
        {
            ll x,y;
            scanf("%lld%lld",&x,&y);
            ll res = query(x,y,1,n,1);
            printf("%lld\n",res);
        }
    }
    return 0;
}

欢迎指出错误qwq

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值