p3966单词

后缀自动机版本：

所有的串用(char)('z'+1)连起来，然后建自动机。再用原串在自动机上跑。跑到的位置的endpos就是出现的次数。不过内存有点大。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <vector>
#include <iomanip>
#include <cstring>
#include <map>
#include <queue>
#include <set>
#include <cassert>
#include <stack>
#include <bitset>
#define mkp make_pair
using namespace std;
const double EPS=1e-12;
typedef long long lon;
const lon SZ=2000010,SSZ=2*SZ,APB=27,one=1,INF=0x7FFFFFFF,mod=1000000007;
int n,len,cnt,slink[SSZ],nex[SSZ][APB],maxlen[SSZ];
int minlen[SSZ],endpos[SSZ],in[SSZ];
char ch[SZ];
string str[SZ];
bool green[SSZ];

void add(string &x)
{
    for(int i=0;i<x.size();++i)
    {
        ch[++len]=x[i];
    }
    ch[++len]='z'+1;
}

int ins(int pre,int c)
{
    int z=++cnt,u=pre;
    green[z]=1;
    for(;u!=-1&&!nex[u][c];u=slink[u])nex[u][c]=z;
    maxlen[z]=maxlen[pre]+1;
    if(u==-1)
    {
        slink[z]=0;
    }
    else
    {
        int x=nex[u][c];
        if(maxlen[x]==maxlen[u]+1)
        {
            slink[z]=x;
        }
        else
        {
            int v=++cnt;
            memcpy(nex[v],nex[x],sizeof(nex[x]));
            maxlen[v]=maxlen[u]+1;
            slink[v]=slink[x];
            minlen[v]=maxlen[slink[v]]+1;
            slink[x]=slink[z]=v;
            minlen[x]=maxlen[v]+1;
            for(;u!=-1&&nex[u][c]==x;u=slink[u])
            nex[u][c]=v;
        }
    }
    minlen[z]=maxlen[slink[z]]+1;
    return z;
}

void topo()
{
    for(int i=1;i<=cnt;++i)
    {
        ++in[slink[i]];
    }
    stack<int> stk;
    for(int i=1;i<=cnt;++i)
    {
        if(!in[i])stk.push(i),endpos[i]=1;
    }
    for(;stk.size();)
    {
        int top=stk.top();
        stk.pop();
        int pre=slink[top];
        endpos[pre]+=endpos[top],--in[pre];
        if(!in[pre]&&pre)
        {
            if(green[pre])++endpos[pre];
            stk.push(pre);
        }
    }
}

void init()
{
    cin>>n;
    for(int i=1;i<=n;++i)
    {
        cin>>str[i];
        add(str[i]);
    }
    int pre=0;
    slink[0]=-1;
    for(int i=1;i<=len;++i)
    {
        pre=ins(pre,ch[i]-'a');
    }
    topo();
}

int calc(string &x)
{
    int cur=0;
    for(int i=0;i<x.size();++i)
    {
        int c=x[i]-'a';
        cur=nex[cur][c];
    }
    return endpos[cur];
}

void work()
{
    for(int i=1;i<=n;++i)
    {
        cout<<calc(str[i])<<endl;
    }
}

int main()
{
    std::ios::sync_with_stdio(0);
    //freopen("d:\\1.txt","r",stdin);
    int casenum;
    //cin>>casenum;
    //cout<<casenum<<endl;
    //for(int time=1;time<=casenum;++time)
    //for(int time=1;cin>>n>>qnum,n;++time)
    {
        //cout<<"Case "<<time<<": ";
        init();
        work();
    }
    return 0;
}