【BZOJ3473&BZOJ3277】字符串（广义后缀自动机）

题意：给定n个字符串，询问每个字符串有多少子串（不包括空串）是所有n个字符串中至少k个字符串的子串？

本质相同的子串算多个。

对于 100% 的数据，1<=n，k<=10^5，所有字符串总长不超过10^5，字符串只包含小写字母。

思路：From 15年国家集训队张天扬论文

 #include<bits/stdc++.h>
 using namespace std;
 typedef long long ll;
 typedef unsigned int uint;
 typedef unsigned long long ull;
 typedef pair<int,int> PII;
 typedef pair<ll,ll> Pll;
 typedef vector<int> VI;
 typedef vector<PII> VII;
 typedef pair<ll,int>P;
 #define N  100010
 #define M  210000
 #define fi first
 #define se second
 #define MP make_pair
 #define pi acos(-1)
 #define mem(a,b) memset(a,b,sizeof(a))
 #define rep(i,a,b) for(int i=(int)a;i<=(int)b;i++)
 #define per(i,a,b) for(int i=(int)a;i>=(int)b;i--)
 #define lowbit(x) x&(-x)
 #define Rand (rand()*(1<<16)+rand())
 #define id(x) ((x)<=B?(x):m-n/(x)+1)
 #define ls p<<1
 #define rs p<<1|1

 const int MOD=1e9+,inv2=(MOD+)/;
       double eps=1e-;
       int INF=<<;
       ll inf=5e13;
       int dx[]={-,,,};
       int dy[]={,,-,};

 string s[N];
 int p,np,q,nq,k,cas;

 int read()
 {
    int v=,f=;
    char c=getchar();
    while(c<||<c) {if(c=='-') f=-; c=getchar();}
    while(<=c&&c<=) v=(v<<)+v+v+c-,c=getchar();
    return v*f;
 }

 struct sam
 {
     int cnt;
     int fa[N<<],t[N<<][];
     int st[N<<],b[N<<],bl[N<<],c[N<<];
     ll f[N<<];

     sam()
     {
         cnt=;
     }

     void add(int x)
     {
         p=np; st[np=++cnt]=st[p]+;
         while(p&&!t[p][x])
         {
             t[p][x]=np;
             p=fa[p];
         }
         if(!p) fa[np]=;
          else if(st[p]+==st[q=t[p][x]]) fa[np]=q;
           else
           {
                   st[nq=++cnt]=st[p]+;
                   memcpy(t[nq],t[q],sizeof t[q]);
                   //t[nq]=t[q];
                   fa[nq]=fa[q];
                   fa[q]=fa[np]=nq;
                   while(p&&t[p][x]==q)
                   {
                       t[p][x]=nq;
                       p=fa[p];
                   }
           }
     }

     void solve()
     {
         rep(i,,cas)
         {
             int u=;
             rep(j,,(int)s[i].size()-)
             {
                 int p=u=t[u][s[i][j]-'a'];
                 while(p!=&&b[p]!=i)
                 {
                     c[p]++; b[p]=i;
                     p=fa[p];
                 }
             }
         }
         rep(i,,cnt) b[i]=;
         rep(i,,cnt) b[st[i]]++;
         rep(i,,cnt) b[i]+=b[i-];
         rep(i,,cnt) bl[b[st[i]]--]=i;

         rep(i,,cnt)
         {
             int p=bl[i];
             if(c[p]>=k) f[p]=f[fa[p]]+st[p]-st[fa[p]];
              else f[p]=f[fa[p]];
         }
         rep(i,,cas)
         {
             ll ans=;
             int u=;
             rep(j,,(int)s[i].size()-)
             {
                 u=t[u][s[i][j]-'a'];
                 ans+=f[u];
             }
             printf("%lld ",ans);
         }

     }
 }sam;

 int main()
 {
     cas=read(),k=read();
     rep(v,,cas)
     {
         cin>>s[v];
         np=;
         rep(i,,(int)s[v].size()-) sam.add(s[v][i]-'a');
     }
     sam.solve();
     return ;
 }