HDU 6194 后缀自动机（2017 ICPC沈阳网络赛A题）

题意

给一个字符串，问出现K次的子串有多少个。

题解

其实是后缀自动机的模板题。。后缀自动机有个right数组，记录的就是从1开始到这个点组成的子串的出现次数。了解了这个以后，从1开始DFS记忆化搜索一下就可以了。

代码

#include <iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<cmath>
#include<queue>
#include<string>
#include<set>
#include<map>
#include<bitset>
#include<stack>
#include<string>
#define UP(i,l,h) for(int i=l;i<h;i++)
#define DOWN(i,h,l) for(int i=h-1;i>=l;i--)
#define W(a) while(a)
#define MEM(a,b) memset(a,b,sizeof(a))
#define LL long long
#define INF 0x3f3f3f3f
#define MAXN 200050
#define MOD 1000000007
#define EPS 1e-3
using namespace std;
char st[MAXN];
int k;
struct SuffixAuto{
    static const int NODE=MAXN<<1,C=26;
    int allc,last,n,par[NODE],len[NODE],trans[NODE][C],c[NODE],id[NODE],rt[NODE],dp[NODE],ans;
    int newNode(){
        int now=++allc;
        MEM(trans[now],0);
        return now;
    }
    void init(){
        MEM(rt,0);
        MEM(c,0);
        MEM(id,0);
        MEM(dp,-1);
        ans=0;
        allc=0;
        last=newNode();
        par[last]=len[last]=0;
    }
    void extend(int c){
        int p=last,np=newNode();
        rt[np]=1;
        len[np]=len[last]+1;
        for(;p&&!trans[p][c];p=par[p]) trans[p][c]=np;
        if(!p) par[np]=1;
        else{
            int q=trans[p][c];
            if(len[q]==len[p]+1) par[np]=q;
            else{
                int nq=++allc;
                par[nq]=par[q],len[nq]=len[p]+1;
                memcpy(trans[nq],trans[q],sizeof(trans[q]));
                par[np]=par[q]=nq;
                for(trans[p][c]=nq,p=par[p];p&&trans[p][c]==q;p=par[p]) trans[p][c]=nq;
            }
        }
        last=np;
    }
    void topSort(){
        UP(i,1,allc+1) c[len[i]]++;
        UP(i,1,n+1) c[i]+=c[i-1];
        UP(i,1,allc+1) id[c[len[i]]--]=i;
        DOWN(i,allc+1,1) rt[par[id[i]]]+=rt[id[i]];
    }
    int dfs(int u){
        if(dp[u]!=-1) return dp[u];
        int sum=0;
        if(rt[u]==k&&u!=1) sum++;
        UP(i,0,C){
            if(trans[u][i]!=0) sum+=dfs(trans[u][i]);
        }
        return dp[u]=sum;
    }
};
SuffixAuto sa;


int main(){
    int t;
    scanf("%d",&t);
    W(t--){
        sa.init();
        scanf("%d",&k);
        scanf("%s",st);
        int n=strlen(st);
        sa.n=n;
        UP(i,0,n) sa.extend(st[i]-'a');
        sa.topSort();
        printf("%d\n",sa.dfs(1));
    }
}