JZOJ4617 【NOI2016模拟7.12】可持久化字符串建Trie后用主席树维护next数组

最新推荐文章于 2024-04-26 18:16:46 发布

原创最新推荐文章于 2024-04-26 18:16:46 发布 · 1.2k 阅读

0 ·

CC 4.0 BY-SA版权

算法-String 同时被 2 个专栏收录

20 篇文章

订阅专栏

算法-主席树

2 篇文章

订阅专栏

本文介绍了一种使用Trie树和主席树解决字符串历史版本查询问题的方法，通过维护节点间的转移关系，实现字符串更新及查找最短循环节长度。

题目大意

现在要求可持久地维护一个字符集大小为 $M$ 字符串 $S$ ，可以支持对历史版本的询问。现在有 $N$ 个操作，对于第 $i$ 个操作包含两个整数 $u,v$ ，表示在第 $u$ 个版本的字符串末尾加上字符 $v$ 作为第 $i$ 个版本的字符串，并且输出每次操作之后的字符串的最短循环节长度。强制在线。

$N,M \leq 3*10^5$

解题思路

首先要知道的是对于一个字符串 $S$ ，它的最小循环节的长度为最大的 $t$ ，使 $t$ 满足既是 $S$ 的前又是 $S$ 的后缀。也就相当于 $KMP$ 的 $next$ 数组，更形象的说一个长度为 $len$ 的的串，它的最小循环节等于 $len-next_{len}$ 。

然后我们发现题目实际上要求我们维护一颗 $Trie$ ，并且对于每条到根的链维护出 $next$ 数组。那么复杂度肯定会有问题。那么我们考虑一样另一种方式维护这个东西。设 $T(x,y)$ 表示在 $x$ 到根的路径上，如果下一个字符为 $y$ 那么根据 $next$ 数组的定义会跳到哪个节点上， $Len_x$ 表示当前这个节点到跟所代表字符串的长度（也就是在 $Trie$ 上的深度）。我们看当前要进行第 $i$ 次操作，插入的字符为 $c$ ，要在第 $u$ 个版本后插入。那么毫无疑问 $Ans =Len_i-Len_{T(u,c)}$ 。那么现在的问题就是怎么维护这个 $T$ 数组。

那么考虑 $T(i)$ 和 $T(T(u,c))$ 的关系。我们设节点 $T(u,c)$ 在节点 $i$ 方向的字符边是 $c'$ ，节点是 $k$ ，那么 $T(i)$ 与 $T(T(u,c))$ 不同的就只有关于 $c'$ 的转移，那么就是说 $T(i) = T(T(u,c))$ ， $T(i,c')=k$ 。这个可以用主席树维护，那么就可以在 $Log$ 的时间转移和维护。

程序

//YxuanwKeith
#include <cstring>
#include <cstdio>
#include <algorithm>

using namespace std;

const int MAXN = 3e5 + 5;

struct Node {
    int l, r, Num;
} Tr[MAXN * 20];

int tot, Ans, N, M, Ord, Fa[MAXN][20], Len[MAXN], Root[MAXN], A[MAXN];

int Query(int Rt, int l, int r, int Side) {
    if (!Rt) return 0;
    if (l == r) return Tr[Rt].Num;
    int Mid = (l + r) >> 1;
    if (Side <= Mid) return Query(Tr[Rt].l, l, Mid, Side); else
        return Query(Tr[Rt].r, Mid + 1, r, Side);
}

void Modify(int &Rt, int rt, int l, int r, int Side, int Poi) {
    Rt = ++ tot;
    Tr[Rt] = Tr[rt];
    if (l == r) {
        Tr[Rt].Num = Poi;
        return;
    }
    int Mid = (l + r) >> 1;
    if (Side <= Mid) Modify(Tr[Rt].l, Tr[rt].l, l, Mid, Side, Poi); else
        Modify(Tr[Rt].r, Tr[rt].r, Mid + 1, r, Side, Poi);
}

int main() {
    scanf("%d%d%d", &N, &M, &Ord);
    for (int i = 1; i <= N; i ++) {
        int u, c;
        scanf("%d%d", &u, &c);
        if (Ord == 1) u ^= Ans, c ^= Ans;
        A[i] = c;
        Fa[i][0] = u;
        for (int j = 1; j < 20; j ++) Fa[i][j] = Fa[Fa[i][j - 1]][j - 1];
        Len[i] = Len[u] + 1;
        int Side = Query(Root[u], 1, M, c);
        Ans = Len[i] - Len[Side];
        printf("%d\n", Ans);
        int p = i, Up = Ans - 1;
        for (int j = 19; j + 1; j --)
            if (Len[i] - Len[Fa[p][j]] <= Up) p = Fa[p][j];
        Modify(Root[i], Root[Side], 1, M, A[p], p);
    }
}