HDU 4641 K-string 2013年多校第4场J题 后缀自动机

题目大意：

就是现在初始的时候给出一个长度为n的串S(n <= 50000), 给定整数K, 然后有m次操作或询问 (m <= 200000), 对于每一次操作或询问, 用1表示在当前的S后面加上一个字母形成新的S, 操作2就是询问当前的字符串中有多少个K串, 当操作是1的时候数字后面会有1个字母表示要添加的字符

K串指的是S的子串, 且在S中出现了不少于K次

大致思路：

首先不难想到后缀自动机, 对于初始的串S建立后缀自动机, 然后每个状态的Right集合的大小是否大于等于K就代表了这个状态代表的子串是否是K串

然后由于后缀自动机的在线性质, 每次添加新的字符到S后面时, 将新的字符插入后缀自动机, 然后考虑新增加的np状态对于Parent书中其父亲结点的Right集合大小的影响即可

写的时候注意一些细节, 刚开始没有注意到nq和q对应状态见得关系WA了几次

代码如下：

Result  :  Accepted     Memory  :  26064 KB     Time  :  1684 ms

/*
 * Author: Gatevin
 * Created Time:  2015/4/13 14:28:04
 * File Name: Rin_Tohsaka.cpp
 */
#include<iostream>
#include<sstream>
#include<fstream>
#include<vector>
#include<list>
#include<deque>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<bitset>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cctype>
#include<cmath>
#include<ctime>
#include<iomanip>
using namespace std;
const double eps(1e-8);
typedef long long lint;
#define foreach(e, x) for(__typeof(x.begin()) e = x.begin(); e != x.end(); ++e)
#define SHOW_MEMORY(x) cout<<sizeof(x)/(1024*1024.)<<"MB"<<endl

#define maxm 250100
#define maxn 500100
#define maxl 50010

int n, K, m;
char s[maxl];
lint ans;

struct Suffix_Automation
{
    struct State
    {
        State *par;
        State *go[26];
        int val, mi, cnt, right;
        void init(int _val = 0)
        {
            val = _val, par = 0, mi = cnt = right = 0;
            memset(go, 0, sizeof(go));
        }
        int calc()
        {
            if(par == 0) return 0;
            else return val - par->val;
        }
    };
    State *root, *last, *cur;
    State nodePool[maxn];
    State* newState(int val = 0)
    {
        cur->init(val);
        return cur++;
    }
    void initSAM()
    {
        cur = nodePool;
        root = newState();
        last = root;
    }
    void extend(int w, bool flag)
    {
        State *p = last;
        State *np = newState(p->val + 1);
        np->right = 1;
        while(p && p->go[w] == 0)
        {
            p->go[w] = np;
            p = p->par;
        }
        if(p == 0)
        {
            np->par = root;
        }
        else
        {
            State *q = p->go[w];
            if(q->val == p->val + 1)
            {
                np->par = q;
            }
            else
            {
                State *nq = newState(p->val + 1);
                memcpy(nq->go, q->go, sizeof(q->go));
                nq->par = q->par;
                if(flag) nq->right = q->right;
                //如果是后来的逐一添加的话就同时更新nq的right
                //不是的话因为会执行topo和SAMInfo函数, 不要更新nq的right
                q->par = nq;
                np->par = nq;
                while(p && p->go[w] == q)
                {
                    p->go[w] = nq;
                    p = p->par;
                }
            }
        }
        last = np;
    }
    
    int d[maxm];
    State *b[maxn];
    void topo()
    {
        int cnt = cur - nodePool;
        int maxVal = 0;
        memset(d, 0, sizeof(d));
        for(int i = 1; i < cnt; i++)
            maxVal = max(maxVal, nodePool[i].val), d[nodePool[i].val]++;
        for(int i = 1; i <= maxVal; i++) d[i] += d[i - 1];
        for(int i = 1; i < cnt; i++) b[d[nodePool[i].val]--] = &nodePool[i];
        b[0] = root;
    }
    void SAMInfo()
    {
        int cnt = cur - nodePool;
        for(int i = cnt - 1; i > 0; i--)
        {
            b[i]->par->right += b[i]->right;
            b[i]->mi = b[i]->par->val + 1;
        }
    }
    void update()//每次新加入结尾的np状态会影响到Parent树上其所有的父亲
    {
        State *np = last;
        while(np != root)
        {
            if(np->right >= K) ans += np->val - np->par->val;
            if(np->par->right >= K) break;//再往上的点没有必要, 因为已经记过数了
            np->par->right++;
            np = np->par;
        }
    }
};

Suffix_Automation sam;

int main()
{
    while(scanf("%d %d %d",&n, &m, &K) != EOF)
    {
        getchar();
        gets(s);
        sam.initSAM();
        for(int i = 0; i < n; i++)
            sam.extend(s[i] - 'a', 0);
        sam.topo();
        sam.SAMInfo();
        ans = 0;
        int cnt = sam.cur - sam.nodePool;
        for(int i = 1; i < cnt; i++)
            if(sam.b[i]->right >= K)
                ans += sam.b[i]->val - sam.b[i]->mi + 1;
        int op;
        char ad;
        while(m--)
        {
            scanf("%d", &op);
            if(op == 2)
                printf("%I64d\n", ans);
            else
            {
                scanf(" %c", &ad);
                sam.extend(ad - 'a', 1);
                sam.update();
            }
        }
    }
    return 0;
}