浅谈可持久化Trie与线段树的原理以及实现(带图)

浅谈可持久化Trie与线段树的原理以及实现

引言

当我们需要保存一个数据结构不同时间的每个版本,最朴素的方法就是每个时间都创建一个独立的数据结构,单独储存。

但是这种方法不仅每次复制新的数据结构需要时间空间上也受不了储存这么多版本的数据结构。

然而有一种叫git的工具,可以维护工程代码的各个版本,而空间上也不至于十分爆炸。怎么做到呢?

答案是版本分支,即每次创建新的版本不完全复制老的数据结构,而是在老的数据结构上加入不同版本的分支。

下面以链表为例

graph LR A-->B B-->C C-->D D-->E E-->F F-->G B-->Z[C_新版本] Z-->Y[D_新版本] Y-->E

新的版本是部分建立在老的版本之上的。不变的地方不变,有编号的地方就加入新版本的分支。

实现可持久化Trie

基于版本分支的思想,我们怎么建立一个可持久化Trie呢?

其次我们注意到在上面那张链表图中进入下一个节点的时候,每次都要判断有没有我们要进入的新版本的分支,十分麻烦。有没有方法可以保证我们向下找的节点全部是我们要的版本呢?

有方法,我们只需要记录修改过的Trie的关键路径就好了。

什么叫关键路径呢?

这里先假设我们只修改Trie上的一个节点。而所谓关键路径就是从Trie的根节点到修改节点的路径,我们只创建这路径上的节点,其余节点全部继承一个老版本的节点。

如图

graph TD ROOT-.c.->c ROOT-.m.->m c--a-->ca ca--t-->cat m--a-->ma ma--p-->map ROOT_NEW--c-->c; ROOT_NEW--m-->m_NEW m_NEW--a-->ma_NEW ma_NEW--p-->map ma_NEW--r-->mar_NEW mar_NEW--k-->mark_NEW

这是一个向有{cat,map}的Trie里插入mark的新单词的例子。

不难发现,在ROOT_NEW可以到达的节点构成的树中,凡是不在mark这个单词的路径上的节点统统用的是老版本树的节点。

代码实现

由于可持久化Trie不是我们的主题,代码就不放了

代码很简单,就不多做解释了

#include <iostream>
#include <string>
using namespace std;

const int N = 1e1 + 128;

int his[128];
int h;

int to[N][26];
int p;

void insert(string &s)
{
    int old = p; //老树的节点
    his[++h] = ++p;
    int now = p; //新树的
    for (auto i : s)
    {
        for (int j = 0; j < 26; j++)
            if (i - 'A' != j) //非关键路径上的节点继承老树
                to[now][j] = to[old][j];
        to[now][i - 'A'] = ++p; //关键路径上的节点就新建
        now = p;
        old = to[now][i - 'A']; //老树也要跟下去
    }
    return;
}

bool ask(int h, string &s) //询问某个版本的Trie里,是否有对应的单词
{
    int now = his[h];
    for (auto i : s)
    {
        if (to[now][i - 'A'] == 0)
            return false;
        now = to[now][i - 'A'];
    }
    return true;
}

int main()
{
    int opt;
    while (cin >> opt)
    {
        if (opt == 1) //插入
        {
            string str;
            cin >> str;
            insert(str);
        }
        else
        {
            string str;
            int h;
            cin >> str >> h;
            cout << ((ask(h, str)) ? 'Y' : 'N') << endl;
        }
    }
    return 0;
}

可持久化线段树

终于到了我们的主题了。可持久化线段树顾名思义就是可持久化的线段树

存在的意义首先是满足部分线段树的要求,然后也能根据线段树的特性解决一部分可持久化Trie的弊端。

聪明的小伙伴可以发现在一个Trie中,我们要把一条完整的子链完全复制下来,如果我们老版本的Trie本来就是一条链,这种操作无异于把Trie重新复制一遍,还是相当慢,怎么办?

在维护一个Trie的时候,这种问题可能会让人头疼。但是线段树可以完全避免,因为线段树的树高是完全有限的[\(Log (n)\)级别]。

基本思路还是只记录修改过的关键路径,不在关键路径上的节点继承老版本的子树。

基本原理还是和可持久化Trie差不多,看图和代码基本也能理解了

graph TD A([1->4]).->B([1->2]) A.->C([3->4]) B-->D([1]) B-->E([2]) C.->F([3]) C.->G([4]) H([1->4_new])-->B H-->I([3->4_new]) I-->F I-->J([4_new])

代码实现

有一点要注意的是,这个线段树的节点关系已经是一个有向图了,不能用满二叉树的性质去计算他的左右儿子。需要手动记录。

其实就是P3919 【模板】可持久化线段树 1(可持久化数组)的题解

#include <iostream>
using namespace std;

const int N = 5e7 + 128;

int num[N / 10];

int his[N / 10];
int h_ptr;

int lc[N], rc[N], val[N];
int p;

int n, m;

void build(int u, int l, int r) //和常规的线段树建立差不多,就是要左右儿子不能用满二叉树性质算出来了,所以要手动存
{
    if (l == r)
    {
        val[u] = num[l];
        return;
    }
    lc[u] = ++p;
    rc[u] = ++p;
    int mid = (l + r) >> 1;
    build(lc[u], l, mid);
    build(rc[u], mid + 1, r);
}

void fork_only(int h) //仅仅只复制一个历史版本
{
    his[h_ptr++] = his[h];
}

void fork_and_edit(int h, int addr, int val_) //复制一个带修改的版本为h的历史版本到最新版本中(把addr这里的数修改为val)
{
    int old = his[h];
    int now = ++p;
    his[h_ptr++] = p;
    int l = 1, r = n;
    while (l < r)
    {
        int mid = (l + r) >> 1;
        if (addr <= mid) //关键路径在左儿子
        {
            rc[now] = rc[old]; //所以右儿子直接继承
            lc[now] = ++p;     //新建一个左儿子
            now = lc[now];
            old = lc[old];
            r = mid;
        }
        else
        {
            lc[now] = lc[old]; //反之继承左儿子
            rc[now] = ++p;
            now = rc[now];
            old = rc[old];
            l = mid + 1;
        }
    }
    val[now] = val_;
    return;
}

int query(int u, int addr)
{
    int l = 1, r = n;
    while (l < r)
    {
        int mid = (l + r) >> 1;
        if (addr <= mid)
            u = lc[u], r = mid;
        else
            u = rc[u], l = mid + 1;
    }
    return val[u];
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    cin >> n >> m;
    for (int i = 1; i <= n; i++)
        cin >> num[i];
    his[h_ptr++] = ++p;
    build(p, 1, n);
    for (int i = 1; i <= m; i++)
    {
        int v, opt, loc;
        cin >> v >> opt >> loc;
        if (opt == 1)
        {
            int value;
            cin >> value;
            fork_and_edit(v, loc, value);
        }
        else
        {
            cout << query(his[v], loc) << endl;
            fork_only(v);
        }
    }
    return 0;
}

如果对代码有问题欢迎评论斧正。

posted @ 2021-09-05 10:08  Icys  阅读(333)  评论(0编辑  收藏  举报