BZOJ 3881 COCI 2015 Divljak

题面

Description

Tom有n个字符串S1,S2...Sn，Jerry有一个字符串集合T，一开始集合是空的。
接下来会发生q个操作，操作有两种形式：
“1 P”，Jerry往自己的集合里添加了一个字符串P。
“2 x”，Tom询问Jerry，集合T中有多少个字符串包含串Sx。（我们称串A包含串B，当且仅当B是A的子串）
Jerry遇到了困难，需要你的帮助。

Input

第1行，一个数n；
接下来n行，每行一个字符串表示Si；
下一行，一个数q；
接下来q行，每行一个操作，格式见题目描述。

Output

对于每一个Tom的询问，帮Jerry输出答案。

Sample Input

3
a
bc
abc
5
1 abca
2 1
1 bca
2 2
2 3

Sample Output

1
2
1

HINT

100%: \(1≤N,Q≤100000\), \(\sum_{s \in S} |s|, \sum_{s \in T}|s| \le 2 \times 10^6\), \(alphabet = \{a ... z\}\)

Solution

正解貌似是fail树加树链的并, 但我写的是后缀树. 还没读入完空间就炸了... 先放一份代码吧

#include <cstdio>
#include <cstring>
#include <vector>
#define vector std::vector

const int LEN = (int)2e6, N = (int)1e5, Q = (int)1e5;
int q;
int ans[Q];
struct segmentTree
{
    struct node
    {
        node *suc[2];
        int sz;
        inline node() {for(int i = 0; i < 2; ++ i) suc[i] = NULL; sz = 0;}
    }*rt;
    inline segmentTree() {rt = NULL;}
    node* insert(node *u, int L, int R, int pos)
    {
        if(u == NULL) u = new node;
        if(L == R) {u->sz = 1; return u;}
        if(pos <= L + R >> 1) u->suc[0] = insert(u->suc[0], L, L + R >> 1, pos); else u->suc[1] = insert(u->suc[1], (L + R >> 1) + 1, R, pos);
        u->sz = 0;
        for(int i = 0; i < 2; ++ i) if(u->suc[i] != NULL) u->sz += u->suc[i]->sz;
        return u;
    }
    inline void insert(int pos) {rt = insert(rt, 1, q, pos);}
    int query(node *u, int L, int R, int pos)
    {
        if(u == NULL) return 0;
        if(R <= pos) return u->sz;
        return query(u->suc[0], L, L + R >> 1, pos) + (pos > L + R >> 1 ? query(u->suc[1], (L + R >> 1) + 1, R, pos) : 0);
    }
    inline int query(int pos) {return query(rt, 1, q, pos);}
    node* merge(int L, int R, node *u, node *_u)
    {
        if(u == NULL) return _u; if(_u == NULL) return u;
        if(L == R) {u->sz |= _u->sz; delete _u; return u;}
        u->suc[0] = merge(L, L + R >> 1, u->suc[0], _u->suc[0]); u->suc[1] = merge((L + R >> 1) + 1, R, u->suc[1], _u->suc[1]);
        delete _u;
        u->sz = 0;
        for(int i = 0; i < 2; ++ i) if(u->suc[i] != NULL) u->sz += u->suc[i]->sz;
        return u;
    }
    inline void merge(segmentTree *T) {rt = merge(1, q, rt, T->rt);}
};
struct suffixAutomaton
{
    struct node
    {
        node *suc[26], *pre;
        int len;
        vector<int> bck, qry;
        int vst;
        vector<node*> successorOnSuffixTree;
        inline node() {for(int i = 0; i < 26; ++ i) suc[i] = NULL; pre = NULL; vst = 0; bck.clear(); qry.clear(); successorOnSuffixTree.clear();}
    }*rt;
    inline suffixAutomaton() {rt = new node; rt->len = 0;}
    inline node* insert(char *str, int len, int id)
    {
        node *lst = rt;
        for(int i = 0; i < len; ++ i)
        {
            int c = str[i] - 'a';
            if(lst->suc[c] != NULL)
            {
                node *p = lst->suc[c];
                if(p->len == lst->len + 1) { if(~ id) p->bck.push_back(id); lst = p; continue;}
                node *q = new node; if(~ id) q->bck.push_back(id);
                for(int i = 0; i < 26; ++ i) q->suc[i] = p->suc[i]; q->pre = p->pre; q->len = lst->len + 1;
                p->pre = q;
                for(; lst != NULL && lst->suc[c] == p; lst = lst->pre) lst->suc[c] = q;
                lst = q;
                continue;
            }
            node *u = new node; u->len = lst->len + 1; if(~ id) u->bck.push_back(id);
            for(; lst != NULL && lst->suc[c] == NULL; lst = lst->pre) lst->suc[c] = u;
            if(lst == NULL) u->pre = rt;
            else
            {
                node *p = lst->suc[c];
                if(p->len == lst->len + 1) u->pre = p;
                else
                {
                    node *q = new node; for(int i = 0; i < 26; ++ i) q->suc[i] = p->suc[i]; q->pre = p->pre; q->len = lst->len + 1;
                    u->pre = p->pre = q;
                    for(; lst != NULL && lst->suc[c] == p; lst = lst->pre) lst->suc[c] = q;
                }
            }
            lst = u;
        }
        return lst;
    }
    void build(node *u)
    {
        u->vst = 1; if(u->pre != NULL) u->pre->successorOnSuffixTree.push_back(u);
        for(int i = 0; i < 26; ++ i) if(u->suc[i] != NULL && ! u->suc[i]->vst) build(u->suc[i]); 
    }
    inline void buildSuffixTree() {build(rt);}
    segmentTree* DFS(node *u)
    {
        segmentTree *seg = new segmentTree;
        for(auto id : u->bck) seg->insert(id);
        for(auto v : u->successorOnSuffixTree)
        {
            segmentTree *res = DFS(v);
            seg->merge(res);
        }
        for(auto qry : u->qry) ans[qry] = seg->query(qry);
        return seg;
    }
    inline void work() {DFS(rt);}
}SAM;
int main()
{
    
    #ifndef ONLINE_JUDGE
    
    freopen("davljak.in", "r", stdin);
    freopen("davljak.out", "w", stdout);
    
    #endif
    
    int n; scanf("%d\n", &n);
    static char str[LEN];
    static suffixAutomaton::node *ed[N + 1];
    for(int i = 1; i <= n; ++ i) scanf("%s", str), ed[i] = SAM.insert(str, strlen(str), -1);
    scanf("%d", &q);
    for(int i = 1; i <= q; ++ i)
    {
        int opt; scanf("%d ", &opt);
        if(opt == 1) scanf("%s", str), SAM.insert(str, strlen(str), i);
        else if(opt == 2)
        {
            int x; scanf("%d", &x);
            ed[x]->qry.push_back(i);
        }
    } 
    SAM.buildSuffixTree();
    memset(ans, -1, sizeof(ans));
    SAM.work();
    for(int i = 1; i <= q; ++ i) if(~ ans[i]) printf("%d\n", ans[i]);
}

转载于:https://www.cnblogs.com/ZeonfaiHo/p/7439631.html