BZOJ 3277&&3473 广义后缀自动机

两道题一模一样，不说什么了。

对于所有串建立广义后缀自动机，用set维护每个状态出现在哪些串里，构建出parent树结构，进行一遍dfs 逆序启发式合并set即可。

统计答案只要在自动机上跑一遍如果k

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<set>
#include<vector>
#define ll long long
using namespace std;
const int maxn = 200010;
set<int> d[maxn];
set<int> :: iterator it;
vector<int> f[maxn];
string s[maxn];
int sum[maxn],n,k;
struct sam
{
    int last,cnt;
    int fa[maxn],mx[maxn],a[maxn][27];
    sam(){
        last=++cnt;
    }
    void insert(int c,int id)
    {
        int p=last,np=last=++cnt;
        mx[np]=mx[p]+1;d[np].insert(id);
        while(!a[p][c]&&p) a[p][c]=np,p=fa[p];
        if(!p) fa[np]=1;
        else
        {
            int q=a[p][c];
            if(mx[p]+1==mx[q]) fa[np]=q;
            else
            {
                int nq=++cnt;mx[nq]=mx[p]+1;
                memcpy(a[nq],a[q],sizeof(a[q]));
                fa[nq]=fa[q];
                fa[np]=fa[q]=nq;
                while(a[p][c]==q) a[p][c]=nq,p=fa[p];
            }
        }
    }
}sam;
void dfs(int u)
{
    for(int i=0;i<f[u].size();i++)
    {
        int v=f[u][i];
        dfs(v);
        if(d[u].size()<d[v].size()) swap(d[u],d[v]);
        for(it=d[v].begin();it!=d[v].end();it++) d[u].insert(*it);
    }
    sum[u]=d[u].size();
}
int main()
{
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++) 
    {
        cin>>s[i];
        int l1=s[i].length();
        for(int j=0;j<l1;j++) sam.insert(s[i][j]-'a',i);
        sam.last=1; 
    }
    for(int i=1;i<=sam.cnt;i++) if(sam.fa[i]) f[sam.fa[i]].push_back(i);
    dfs(1);
    if(k>n) {
        for(int i=1;i<=n;i++) printf("0 ");
        return 0;
    }
    for(int i=1;i<=n;i++)
    {
        ll ans=0;int now=1;
        for(int j=0;j<s[i].length();j++)
        {
            now=sam.a[now][s[i][j]-'a'];
            while(sum[now]<k&&now) now=sam.fa[now];
            ans+=sam.mx[now];
        }
        printf("%d ",ans);
    }
    return 0;
}