可持久化线段树

在这里,所谓“可持久化”的数据结构并非指将数据存在非易失的存储器上,而是指保存了数据修改的历史信息。比如说对可持久化线段树进行修改操作,操作完成后我们可以在线段树原有的时间复杂度内查询到希望查询的版本的信息,比如“第二次修改后区间L和R之间的和”。

通常遇到的线段树都是构建之后结构不变化的,所以在修改关键值时,只有节点内的值受到影响,而树本身的结构不发生变化(比如左右子节点所表示的区间)。这为线段树进行可持久化提供了便利。我们每次修改的时候不直接改动原来节点的值,而是创建一系列新的节点。如果整棵树复制的话不仅非常耗费时间,而且占用空间太大。在线段树的单次修改中,实际上受到影响的节点是有限的,原来的节点可以得到重复利用。

修改的过程

可持久化线段树每次修改都会自上而下地新建一些节点。每次修改后的版本都有一个根节点与之对应。

只考虑单点修改,我们将递归过程中所有的节点创建一个“影子节点”,所谓“影子节点”保存的是当前修改结束后的受到更改的值。当u是v的影子节点时,我们称v时u的原节点。

对一棵线段树进行的两次可能发生的修改操作,注意到每个根节点都对应了一次修改,不同的修改操作用了不同的颜色进行区分

在线段树的修改操作中,子节点修改完成后只影响到父节点(pushup操作),而不会影响到兄弟节点。所以我们发现,当修改影响到非叶子结点u时,(在单点修改中)他一定只有一个子节点会受到修改的影响,比如右子节点受到影响,此时u的影子节点v的左子节点指向u的左子节点,而v的右子节点对应的是受到影响的新右子节点w,w是u的右子节点的影子节点,以此类推。

Lazy标签

对于区间修改,需要维护一个lazy标签来推迟更新操作,在pushdown操作时,创建了u的原节点的子节点的影子节点,在实际实现中,通过维护一个节点的origin节点指针就可以做到这一点。

红色是一个标记了lazy的操作,操作结束后并没有立刻生成子节点的影子节点。而在pushdown操作时(紫色),新的影子节点的值从原节点修改而来

实现

由于可持久化线段树在修改过程中需要不断新建影子节点,所以通常的下标标记子节点的方法不再有效。节点需要维护的不仅仅是线段树的关键值x,还有左右子节点指针lch、rch,lazy标记和原节点指针origin。

在修改线段树时,沿着最新版本的线段树自上而下地遍历、创建影子节点并修改即可。

下面的程序实现了一个维护区间和的可持久化线段树,支持区间修改。而且这个程序包含了一个demo。首先输入一个n,随后输入n个整数,表示a[1]~a[n]的初始值。随后开始查询和修改操作,输入q查询,m修改。查询接受一个版本号(从0开始),输出序列所有的值。修改接受u、v和w,表示将区间[u,v]每个元素加上w。

#include <bits/stdc++.h>
using namespace std;
// 可持久化线段树
const int N = 100010;
struct Node {
    int sum, lch, rch, lazy, origin;
    Node():sum(0),lch(-1),rch(-1),lazy(0),origin(-1) {}
}tree[(N<<2)*4];
int tot, a[N], rt[N], curver;

void init() { tot=0; curver=0; }
int createIndentity(int p) {    // 创建影子节点
    int cp=tot++;
    tree[cp]=Node();
    tree[cp].origin=p;tree[cp].sum=tree[p].sum;tree[cp].lazy=tree[p].lazy;
    return cp;
}
void pushup(int p) { tree[p].sum=tree[tree[p].lch].sum+tree[tree[p].rch].sum; }
void pushdown(int p,int l,int r,int m) {
    int lch=tree[p].lch, rch=tree[p].rch;
    if (lch==-1||rch==-1) {
        int o=tree[p].origin;
        lch=tree[p].lch=createIndentity(tree[o].lch);
        rch=tree[p].rch=createIndentity(tree[o].rch);
    }
    tree[lch].lazy+=tree[p].lazy; tree[rch].lazy+=tree[p].lazy;
    tree[lch].sum+=tree[p].lazy*(m-l+1); tree[rch].sum+=tree[p].lazy*(r-m);
    tree[p].lazy=0;
}
int build(int l, int r) {
    int p=tot++;
    tree[p]=Node();
    tree[p].sum=a[l];
    if (l==r) return p;
    int mid=l+r>>1;
    tree[p].lch=build(l,mid);
    tree[p].rch=build(mid+1,r);
    pushup(p);
    return p;
}
int add(int p, int l, int r, int x, int y, int z) {
    int cp=tot++;       // create shadow node
    tree[cp]=Node();
    tree[cp].origin=p;  // origin node number, prepared for pushdown operation
    if (x<=l&&r<=y){
        tree[cp].lazy=tree[p].lazy+z;
        tree[cp].sum=tree[p].sum+z*(r-l+1);
        return cp;
    }
    int mid=l+r>>1;
    if (tree[p].lazy) pushdown(p,l,r,mid);
    if (x<=mid)
        tree[cp].lch=add(tree[p].lch,l,mid,x,y,z);
    else tree[cp].lch=tree[p].lch;
    if (mid<y)
        tree[cp].rch=add(tree[p].rch,mid+1,r,x,y,z);
    else tree[cp].rch=tree[p].rch;
    pushup(cp);
    return cp;
}
int query(int p, int l, int r, int x, int y) {
    if (x<=l&&r<=y) return tree[p].sum;
    int mid=l+r>>1, ret=0;
    if (tree[p].lazy) pushdown(p,l,r,mid);
    if (x<=mid) ret+=query(tree[p].lch,l,mid,x,y);
    if (mid<y) ret+=query(tree[p].rch,mid+1,r,x,y);
    return ret;
}

int main() {
    int n;
    scanf("%d", &n);
    for (int i=1;i<=n;++i) scanf("%d", a+i);
    // 
    init();
    rt[curver]=build(1,n);
    for (;;){
        int u,v,w;
        char q[3];
        printf("q/m:");
        scanf("%s", q);
        if (q[0]=='m') {
            printf("Please input u, v, w and we will add w to [u,v]: ");
            scanf("%d%d%d", &u, &v, &w);
            rt[curver+1]=add(rt[curver],1,n,u,v,w);
            ++curver;
            for (int i=1;i<=n;++i) {
                printf("%d ", query(rt[curver],1,n,i,i));
            }
            puts("");
        }else {
            printf("Please input ver: ");
            scanf("%d", &w);
            for (int i=1;i<=n;++i) {`
                printf("%d ", query(rt[w],1,n,i,i));
            }
            puts("");
        }
    }
    return 0;
}

Have fun!

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 156,907评论 4 360
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 66,546评论 1 289
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 106,705评论 0 238
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,624评论 0 203
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 51,940评论 3 285
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,371评论 1 210
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,672评论 2 310
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,396评论 0 195
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,069评论 1 238
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,350评论 2 242
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 31,876评论 1 256
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,243评论 2 251
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 32,847评论 3 231
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,004评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,755评论 0 192
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,378评论 2 269
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,266评论 2 259

推荐阅读更多精彩内容