luogu P5341 [TJOI2019]甲苯先生和大中锋的字符串

传送门

考虑子串以及出现个数,可以发现SAM可以快速知道每种子串的出现次数,即所在状态的\(endpos\)集合大小,然后一个状态对应的子串长度是一段连续区间,所以可以对每个状态差分一下,就能统计答案了

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<vector>
#include<cmath>
#include<ctime>
#include<queue>
#include<map>
#include<set>
#define LL long long
#define db double

using namespace std;
const int N=2e5+10;
/*int rd()
{
    int x=0,w=1;char ch=0;
    while(ch<'0'||ch>'9'){if(ch=='-') w=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+(ch^48);ch=getchar();}
    return x*w;
}*/ //垃圾luogu数据卡读优
int to[N],nt[N],hd[N],tot;
void add(int x,int y)
{
    ++tot,to[tot]=y,nt[tot]=hd[x],hd[x]=tot;
}
char cc[N];
int n,kk,b[N],la,tt,fa[N],ch[N][26],len[N],sz[N];
int newnode(){++tt,fa[tt]=len[tt]=sz[tt]=0;memset(ch[tt],0,sizeof(int)*26);return tt;}
void extd(int x)
{
    int np=newnode(),p=la;
    len[np]=len[p]+1,sz[np]=1,la=np;
    while(p&&!ch[p][x]) ch[p][x]=np,p=fa[p];
    if(!p) fa[np]=1;
    else
    {
        int q=ch[p][x];
        if(len[q]==len[p]+1) fa[np]=q;
        else
        {
            int nq=newnode();
            fa[nq]=fa[q],len[nq]=len[p]+1,memcpy(ch[nq],ch[q],sizeof(ch[q]));
            fa[np]=fa[q]=nq;
            while(p&&ch[p][x]==q) ch[p][x]=nq,p=fa[p];
        }
    }
}
void dfs(int x)
{
    for(int i=hd[x];i;i=nt[i])
        dfs(to[i]),sz[x]+=sz[to[i]];
    if(sz[x]==kk) ++b[len[fa[x]]+1],--b[len[x]+1];
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%s",cc+1);
        n=strlen(cc+1);
        scanf("%d",&kk);
        for(int i=1;i<=n;++i) b[i]=0;
        la=1,tt=0,newnode();
        for(int i=1;i<=n;++i) extd(cc[i]-'a');
        memset(hd,0,sizeof(int)*(tt+3)),tot=0;
        for(int i=2;i<=tt;++i) add(fa[i],i);
        dfs(1);
        int ans=-1,ma=0;
        for(int i=1;i<=n;++i) b[i]+=b[i-1];
        for(int i=n;i;--i)
            if(ma<b[i]) ma=b[i],ans=i;
        printf("%d\n",ans);
    }
    return 0;
}