bzoj 4180: 字符串计数 后缀自动机+倍增floyd

题意

SD有一名神犇叫做Oxer，他觉得字符串的题目都太水了，于是便出了一道题来虐蒟蒻yts1999。
他给出了一个字符串T，字符串T中有且仅有4种字符 ‘A’, ‘B’, ‘C’, ‘D’。现在他要求蒟蒻yts1999构造一个新的字符串S，构造的方法是：进行多次操作，每一次操作选择T的一个子串，将其加入S的末尾。
对于一个可构造出的字符串S，可能有多种构造方案，Oxer定义构造字符串S所需的操作次数为所有构造方案中操作次数的最小值。
Oxer想知道对于给定的正整数N和字符串T，他所能构造出的所有长度为N的字符串S中，构造所需的操作次数最大的字符串的操作次数。
蒟蒻yts1999当然不会做了，于是向你求助。
1 ≤ N ≤ 10^18，1 ≤ |T| ≤ 10^5。

分析

我们要是操作数尽量大，就是说我们每次操作都要保证加入的子串是极大的，也就是不能和后一个加入或前一个加入的子串构成更长的子串，且子串长度要尽量小。
那么可以用后缀自动机来求出a[i,j]表示以i开头最短的子串满足该子串后面接j后产生的串不是T的子串。
然后二分答案，跑一下倍增floyd验证就好了。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long LL;

const int N=100005;
const LL inf=(LL)2e18;

int sz,last,ch[N*2][4],fa[N*2],mx[N*2],r[N*2],deg[N*2];
LL n,f[N*2];
char str[N];
struct Matrix {LL a[4][4];}a;

void ins(int x)
{
    int p,q,np,nq;
    p=last;last=np=++sz;mx[np]=mx[p]+1;
    for (;p&&!ch[p][x];p=fa[p]) ch[p][x]=np;
    if (!p) fa[np]=1;
    else
    {
        q=ch[p][x];
        if (mx[q]==mx[p]+1) fa[np]=q;
        else
        {
            nq=++sz;mx[nq]=mx[p]+1;
            memcpy(ch[nq],ch[q],sizeof(ch[q]));
            fa[nq]=fa[q];
            fa[q]=fa[np]=nq;
            for (;ch[p][x]==q;p=fa[p]) ch[p][x]=nq;
        }
    }
}

void topusort()
{
    for (int i=1;i<=sz;i++)
        for (int j=0;j<4;j++)
            if (ch[i][j]) deg[ch[i][j]]++;
    int tot=0;
    for (int i=1;i<=sz;i++) if (!deg[i]) r[++tot]=i;
    for (int i=1;i<=tot;i++)
        for (int j=0;j<4;j++)
        {
            deg[ch[r[i]][j]]--;
            if (!deg[ch[r[i]][j]]) r[++tot]=ch[r[i]][j];
        }
}

void pre(int now)
{
    for (int i=1;i<=sz;i++) f[i]=inf;
    f[ch[1][now]]=1;
    for (int i=1;i<=sz;i++)
    {
        int x=r[i];
        for (int j=0;j<4;j++)
            if (!ch[x][j]) a.a[now][j]=min(a.a[now][j],f[x]);
            else f[ch[x][j]]=min(f[ch[x][j]],f[x]+1);
    }
}

void mul(Matrix a,Matrix b,Matrix &c)
{
    for (int i=0;i<4;i++)
        for (int j=0;j<4;j++)
            c.a[i][j]=inf;
    for (int i=0;i<4;i++)
        for (int j=0;j<4;j++)
            for (int k=0;k<4;k++)
                c.a[i][j]=min(c.a[i][j],a.a[i][k]+b.a[k][j]);
}

bool check(LL y)
{
    Matrix ans=a,x=a;y--;
    while (y)
    {
        if (y&1) mul(ans,x,ans);
        mul(x,x,x);y>>=1;
    }
    LL mn=inf;
    for (int i=0;i<4;i++)
        for (int j=0;j<4;j++)
            mn=min(mn,ans.a[i][j]);
    return mn+1<=n;
}

int main()
{
    scanf("%lld%s",&n,str);
    int len=strlen(str);
    sz=last=1;
    for (int i=0;i<len;i++) ins(str[i]-'A');
    topusort();
    for (int i=0;i<4;i++)
    {
        for (int j=0;j<4;j++) a.a[i][j]=inf;
        pre(i);
    }
    LL l=1,r=n;
    while (l<=r)
    {
        LL mid=(l+r)/2;
        if (check(mid)) l=mid+1;
        else r=mid-1;
    }
    printf("%lld",l);
    return 0;
}