2016-2017 ACM-ICPC Southeastern European Regional Programming Contest (SEERC 2016) C Castle (kmp)

题目链接：2016-2017 ACM-ICPC Southeastern European Regional Programming Contest (SEERC 2016) C Castle

题意：

初始长为 N 的串 S，以及一个空集合 T。下列有 E 个操作:

1 c ：在 S 串尾添加字符 c

2 ：将当前串 S 加入到集合 T 中

3 ：要求判断 T 中多少个串是当前串 S 的后缀。

题解：

题意可以转换为有多少个标记的前缀是当前字符串的后缀。

那么我们可以想到kmp的next数组，next数组记录的值就是当前字符串后缀与前缀的最大匹配。

那么通过这个信息就可以推出有多少个标记的前缀是当前字符串的后缀。

#include<bits/stdc++.h>
#define F(i,a,b) for(int i=(a);i<=(b);++i)
using namespace std;

const int N=1200007;

char a[N];
int n,m,nxt[N],cnt[N],i,j,op,ed;
char o[2];

int main(){
    scanf("%d%d%s",&n,&m,a);
    for (nxt[0] = j = -1, i = 1; i < n; nxt[i++] = j) {
        while (~j&&a[j + 1] != a[i])j = nxt[j];
        if (a[j + 1] == a[i])j++;
    }
    F(ii,1,m)
    {
        scanf("%d",&op);
        if(op==1)
        {
            scanf("%s",o);
            a[n++]=*o;
            while (~j&&a[j + 1] != a[i])j = nxt[j];
            if (a[j + 1] == a[i])j++;
            nxt[i]=j,cnt[i]=cnt[j],i++;
        }else if(op==2)cnt[n-1]+=ed!=n,ed=n;
        else printf("%d\n",cnt[n-1]);
    }
    return 0;
}