hdoj5769后缀自动机版本

网上的题解都是后缀数组，我来个后缀自动机题解。

建好后缀自动机后由于后缀自动机是单向的，那么dfs一遍记录各节点的size,要保证一个节点只经过一次才是O(n)，否则是O(n^2)。表示这个节点及后面还有几个节点。然后再来个ans数组，再dfs一次。这次如果走的是题目要的字母（记c),那么ans[x]+=siz[to]，因为to能到的节点对应的子串都有c。如果走的不是c,那么ans[x]+=ans[to]。ans其实就表示该节点以后能有几个能有c的子串。同样ans只能走一次，否则是n^2的。那么开始时memset为-1。走到一个节点就赋为0。这样-1的点就是还需要dfs的，一个点只走一次。具体看两个dfs函数。

#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-;
typedef long long lon;
const lon SZ=,SSZ=*SZ,APB=,INF=0x7FFFFFFF,mod=;
lon cnt,maxlen[SSZ],minlen[SSZ],nex[SSZ][APB];
lon slink[SSZ],siz[SSZ],ans[SSZ];
char dst,ch[SZ];

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

void init()
{
    scanf(" %c",&dst);
    scanf(" %s",ch+);
    lon pre=;
    slink[]=-;
    cnt=;
    memset(ans,-,sizeof(ans));
    for(lon i=;ch[i];++i)
    {
        pre=add(pre,ch[i]-'a');
    }
}

void dfs1(lon x)
{
    siz[x]=;
    for(lon i=;i<APB;++i)
    {
        lon t=nex[x][i];
        if(t)
        {
            if(!siz[t])dfs1(t);
            siz[x]+=siz[t];
        }
    }
}

void dfs2(lon x)
{
    ans[x]=;
    for(lon i=;i<APB;++i)
    {
        lon t=nex[x][i];
        if(t)
        {
            if(ans[t]==-)dfs2(t);
            if(i==dst-'a')ans[x]+=siz[t];
            else ans[x]+=ans[t];
        }
    }
}

void work()
{
    dfs1();
    dfs2();
    cout<<ans[]<<endl;
    for(int i=;i<=cnt;++i)
    {
        memset(nex[i],,sizeof(nex[i]));
        siz[i]=;
    }
}

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