线段树建造

1、由于二叉树的自身特性,对于每个父亲节点的编号 i,他的两个儿子的编号分别是 2i 和 2i+1,所以我们考虑写两个 O(1) 的取儿子函数:

inline int ls(int p){return p<<1;} 左儿子 inline int rs(int p){return p<<1|1;} 右儿子 < code></1;}>

二进制位左移一位代表着数值 X 2,而如果左移完之后再或上 1,由于左移完之后最后一位二进制位上一定会是 0 ,所以 |1 等价于 +1 。这个地方更多是为了方便,速度方面理论上是要比 +1 快,但其实编译器会帮你主动干这件事。

2、 那么根据线段树的服务对象,可以得到线段树的维护:

void push_up_sum(int p){
    t[p]=t[lc(p)]+t[rc(p)];
}//   &#x5411;&#x4E0A;&#x4E0D;&#x65AD;&#x7EF4;&#x62A4;&#x533A;&#x95F4;&#x64CD;&#x4F5C;

void push_up_min(int p){//max and min
     t[p]=min(t[lc(p)],t[rc(p)]);
     //t[p]=max(t[lc(p)],t[rc(p)]);
}

此处一定要注意,push up 操作的目的是为了维护父子节点之间的逻辑关系。当我们递归建树时,对于每一个节点我们都需要遍历一遍,并且电脑中的递归实际意义是先向底层递归,然后从底层向上回溯,所以开始递归之后必然是先去整合子节点的信息,再向它们的祖先回溯整合之后的信息。(这其实是正确性的证明啦)。

呐,我们在这儿就能看出来,实际上 push_up 是在合并两个子节点的信息,所以需要信息满足结合律!

3、 那么对于建树,由于二叉树自身的父子节点之间的可传递关系,所以可以考虑递归建树 ,并且在建树的同时,我们应该维护父子节点的关系:

int n;
int ans[MAXN*4];

void build(ll p,ll l,ll r){//p&#x4E3A;&#x6839;&#x8282;&#x70B9;&#xFF0C;l&#x4E3A;&#x533A;&#x95F4;&#x5F00;&#x59CB;&#x8282;&#x70B9;&#xFF0C;r&#x4E3A;&#x533A;&#x95F4;&#x7ED3;&#x675F;&#x8282;&#x70B9;
  if(l==r){ans[p]=a[l];return ;}
  //&#x5982;&#x679C;&#x5DE6;&#x53F3;&#x533A;&#x95F4;&#x76F8;&#x540C;&#xFF0C;&#x90A3;&#x4E48;&#x5FC5;&#x7136;&#x662F;&#x53F6;&#x5B50;&#x8282;&#x70B9;&#x5566;&#xFF0C;&#x53EA;&#x6709;&#x53F6;&#x5B50;&#x8282;&#x70B9;&#x662F;&#x88AB;&#x771F;&#x5B9E;&#x8D4B;&#x503C;&#x7684;
  ll mid=(l+r)>>1;
  build(ls(p),l,mid);
  build(rs(p),mid+1,r);
//&#x6B64;&#x5904;&#x7531;&#x4E8E;&#x6211;&#x4EEC;&#x91C7;&#x7528;&#x7684;&#x662F;&#x4E8C;&#x53C9;&#x6811;&#xFF0C;&#x6240;&#x4EE5;&#x5BF9;&#x4E8E;&#x6574;&#x4E2A;&#x7ED3;&#x6784;&#x6765;&#x8BF4;&#xFF0C;&#x53EF;&#x4EE5;&#x7528;&#x4E8C;&#x5206;&#x6765;&#x964D;&#x4F4E;&#x590D;&#x6742;&#x5EA6;&#xFF0C;&#x5426;&#x5219;&#x6811;&#x5F62;&#x7ED3;&#x6784;&#x5219;&#x6CA1;&#x6709;&#x4EC0;&#x4E48;&#x660E;&#x663E;&#x7684;&#x4F18;&#x5316;
  push_up(p);
//&#x6B64;&#x5904;&#x7531;&#x4E8E;&#x6211;&#x4EEC;&#x662F;&#x8981;&#x901A;&#x8FC7;&#x5B50;&#x8282;&#x70B9;&#x6765;&#x7EF4;&#x62A4;&#x7236;&#x4EB2;&#x8282;&#x70B9;&#xFF0C;&#x6240;&#x4EE5;pushup&#x7684;&#x4F4D;&#x7F6E;&#x5E94;&#x5F53;&#x662F;&#x5728;&#x56DE;&#x6EAF;&#x65F6;&#x3002;&#x6B64;&#x65F6;&#x5B50;&#x8282;&#x70B9;&#x7684;&#x6570;&#x636E;&#x5DF2;&#x7ECF;&#x77E5;&#x9053;&#x3002;
}

那么对于区间操作,我们考虑引入一个名叫” lazy~tag”(懒标记)的东西——之所以称其”lazy”,是因为原本区间修改需要通过先改变叶子节点的值,然后不断地向上递归修改祖先节点直至到达根节点,时间复杂度最高可以到达 O(nlogn) 的级别。但当我们引入了懒标记之后,区间更新的期望复杂度就降到了 O(logn) 的级别且甚至会更低.

1、首先先来从分块思想上解释如何区间修改:
分块的思想是通过将整个序列分为有穷个小块,对于要查询的一段区间,总是可以整合成 k 个所分块与 m 个单个元素的信息的并

那么我们可以反过来思考这个问题:对于一个要修改的、长度为 l 的区间来说,总是可以看做由一个长度为2^log(n)和剩下的元素(或者小区间组成)。

那么我们就可以先将其拆分成线段树上节点所示的区间,之后分开处理:

(如果单个元素被包含就只改变自己,如果整个区间被包含就修改整个区间。)

其实好像这个在分块里不是特别简单地实现,但是在线段树里,无论是元素还是区间都是线段树上的一个节点,所以我们不需要区分区间还是元素,加个判断就好。

2、懒标记的正确打开方式

首先,懒标记的作用是记录每次、每个节点要更新的值,也就是 Δ。但线段树的优点不在于全记录(全记录依然很慢 qwq),而在于传递式记录:
整个区间都被操作,记录在公共祖先节点上;只修改了一部分,那么就记录在这部分的公共祖先上;如果四环以内只修改了自己的话,那就只改变自己。

After that,如果我们采用上述的优化方式的话,我们就需要在每次区间的查询修改时 push_down 一次,以免重复或者冲突或者爆炸
那么对于 push_down 而言,其实就是纯粹的 push_up 的逆向思维(但不是逆向操作): 因为修改信息存在父节点上,所以要由父节点向下传导 lazy~tag 。

那么问题来了:怎么传导 push_down 呢?这里很有意思,开始回溯时执行 push_up,因为是向上传导信息;那我们如果要让它向下更新,就调整顺序,在向下递归的时候 push_down 不就好惹~ qwq:

inline void f(ll p,ll l,ll r,ll k){
   tag[p]=tag[p]+k;
   ans[p]=ans[p]+k*(r-l+1);
   //&#x7531;&#x4E8E;&#x662F;&#x8FD9;&#x4E2A;&#x533A;&#x95F4;&#x7EDF;&#x4E00;&#x6539;&#x53D8;&#xFF0C;&#x6240;&#x4EE5;ans&#x6570;&#x7EC4;&#x8981;&#x52A0;&#x5143;&#x7D20;&#x4E2A;&#x6570;&#x6B21;&#x5566;
}
//&#x6211;&#x4EEC;&#x53EF;&#x4EE5;&#x8BA4;&#x8BC6;&#x5230;&#xFF0C;f&#x51FD;&#x6570;&#x7684;&#x552F;&#x4E00;&#x76EE;&#x7684;&#xFF0C;&#x5C31;&#x662F;&#x8BB0;&#x5F55;&#x5F53;&#x524D;&#x8282;&#x70B9;&#x6240;&#x4EE3;&#x8868;&#x7684;&#x533A;&#x95F4;
inline void push_down(ll p,ll l,ll r){
   ll mid=(l+r)>>1;
   f(ls(p),l,mid,tag[p]);
   f(rs(p),mid+1,r,tag[p]);
   tag[p]=0;
   //&#x6BCF;&#x6B21;&#x66F4;&#x65B0;&#x4E24;&#x4E2A;&#x513F;&#x5B50;&#x8282;&#x70B9;&#x3002;&#x4EE5;&#x6B64;&#x4E0D;&#x65AD;&#x5411;&#x4E0B;&#x4F20;&#x9012;
}
inline void update(ll nl,ll nr,ll l,ll r,ll p,ll k){
   //nl,nr&#x4E3A;&#x8981;&#x4FEE;&#x6539;&#x7684;&#x533A;&#x95F4;
   //l,r,p&#x4E3A;&#x5F53;&#x524D;&#x8282;&#x70B9;&#x6240;&#x5B58;&#x50A8;&#x7684;&#x533A;&#x95F4;&#x4EE5;&#x53CA;&#x8282;&#x70B9;&#x7684;&#x7F16;&#x53F7;
   if(nl<=l&&r<=nr) { ans[p]+="k*(r-l+1);" tag[p]+="k;" return ; } push_down(p,l,r); 回溯之前(也可以说是下一次递归之前,因为没有递归就没有回溯) 由于是在回溯之前不断向下传递,所以自然每个节点都可以更新到 ll mid="(l+r)">>1;
   if(nl<=mid)update(nl,nr,l,mid,ls(p),k); if(nr>mid) update(nl,nr,mid+1,r,rs(p),k);
   push_up(p);
   //&#x56DE;&#x6EAF;&#x4E4B;&#x540E;
}
</=mid)update(nl,nr,l,mid,ls(p),k);></=l&&r<=nr)>

代码如下

ll query(ll q_x,ll q_y,ll l,ll r,ll p){
        ll res=0;
        if(q_x<=l&&r<=q_y)return ans[p]; ll mid="(l+r)">>1;
        push_down(p,l,r);
        if(q_x<=mid)res+=query(q_x,q_y,l,mid,ls(p)); if(q_y>mid) res+=query(q_x,q_y,mid+1,r,rs(p));
        return res;
}
</=mid)res+=query(q_x,q_y,l,mid,ls(p));></=l&&r<=q_y)return>

这是所有的函数:

unsigned ll n,m,a[MAXN],ans[MAXN<<2],tag[maxn<<2]; tag数组为标记 inline ll ls(ll x) { return x<<1; } rs(ll x<<1|1; void push_up(ll p) ans[p]="ans[ls(p)]+ans[rs(p)];" build(ll p,ll l,ll r) tag[p]="0;" if(l="=r){ans[p]=a[l];return" ;} mid="(l+r)">>1;
    build(ls(p),l,mid);
    build(rs(p),mid+1,r);
    push_up(p);
}

inline void f(ll p,ll l,ll r,ll k)
{
    tag[p]=tag[p]+k;
    ans[p]=ans[p]+k*(r-l+1);
}

inline void push_down(ll p,ll l,ll r)
{
    ll mid=(l+r)>>1;
    f(ls(p),l,mid,tag[p]);
    f(rs(p),mid+1,r,tag[p]);
    tag[p]=0;
}

inline void update(ll nl,ll nr,ll l,ll r,ll p,ll k)
{
    if(nl<=l&&r<=nr) { ans[p]+="k*(r-l+1);" tag[p]+="k;" return ; } push_down(p,l,r); ll mid="(l+r)">>1;
    if(nl<=mid)update(nl,nr,l,mid,ls(p),k); if(nr>mid) update(nl,nr,mid+1,r,rs(p),k);
    push_up(p);
}

ll query(ll q_x,ll q_y,ll l,ll r,ll p)
{
    ll res=0;
    if(q_x<=l&&r<=q_y)return ans[p]; ll mid="(l+r)">>1;
    push_down(p,l,r);
    if(q_x<=mid)res+=query(q_x,q_y,l,mid,ls(p)); if(q_y>mid) res+=query(q_x,q_y,mid+1,r,rs(p));
    return res;
}
</=mid)res+=query(q_x,q_y,l,mid,ls(p));></=l&&r<=q_y)return></=mid)update(nl,nr,l,mid,ls(p),k);></=l&&r<=nr)></2],tag[maxn<<2];>

Original: https://www.cnblogs.com/aska00/p/16524813.html
Author: Aska0
Title: 线段树建造

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/587935/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球