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

题意

给出一个字符串 \(s\) 和一个整数 \(k\)，问 \(s\) 的所有恰好出现 \(k\) 次字串中长度出现次数最多的串中最长的串的长度，有多测。
\(|n|\le10^5\)。

思路

首先建 \(SAM\)，求出每个点代表的 endpos 等价类的大小，用树状数组维护所有大小为 \(k\) 的等价类中每种长度的字符串出现次数，最后去 \(\max\)。

代码

// Problem: P5341 [TJOI2019] 甲苯先生和大中锋的字符串
// Contest: Luogu
// URL: https://www.luogu.com.cn/problem/P5341
// Memory Limit: 125 MB
// Time Limit: 1000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

#include<bits/stdc++.h>
using namespace std;
namespace IO{
    template<typename T>
    inline void read(T&x){
        x=0;char c=getchar();bool f=0;
        while(!isdigit(c)) c=='-'?f=1:0,c=getchar();
        while(isdigit(c)) x=x*10+c-'0',c=getchar();
        f?x=-x:0;
    }
    template<typename T>
    inline void write(T x){
        if(x==0){putchar('0');return ;}
        x<0?x=-x,putchar('-'):0;short st[50],top=0;
        while(x) st[++top]=x%10,x/=10;
        while(top) putchar(st[top--]+'0');
    }
    inline void read(char&c){c=getchar();while(isspace(c)) c=getchar();}
    inline void write(char c){putchar(c);}
    inline void read(string&s){s.clear();char c;read(c);while(!isspace(c)&&~c) s+=c,c=getchar();}
    inline void write(string s){for(int i=0,len=s.size();i<len;i++) putchar(s[i]);}
    template<typename T>inline void write(T*x){while(*x) putchar(*(x++));}
    template<typename T,typename...T2> inline void read(T&x,T2&...y){read(x),read(y...);}
    template<typename T,typename...T2> inline void write(const T x,const T2...y){write(x),putchar(' '),write(y...),sizeof...(y)==1?putchar('\n'):0;}
}using namespace IO;
const int maxn=100010;
string s;
int k,n;
class BIT{
private:
    int t[maxn];
    int low(int u){return u&-u;}
public:
    void update(int u,int add){while(u<=n) t[u]+=add,u+=low(u);}
    int query(int u){int ans=0;while(u) ans+=t[u],u-=low(u);return ans;}
    void clear(){memset(t,0,sizeof(t));}
}T;
class SAM{
private:
    struct node{int ch[26],link,len,sz;}t[maxn*2];
    int cnt,lt;
    vector<int>e[maxn*2];
public:
    void insert(int c){
        int d=++cnt,u=lt,p=u;
        t[d].len=t[lt].len+1;
        t[d].sz=1;
        lt=d;
        while(u&&t[u].ch[c]==0) t[u].ch[c]=d,p=u=t[u].link;
        if(p==0){t[d].link=1;return ;}
        int q=t[p].ch[c];
        if(t[q].len==t[p].len+1){t[d].link=q;return ;}
        int clone=++cnt;
        t[clone]=t[q],t[clone].sz=0,t[clone].len=t[p].len+1;
        u=p;
        while(u&&t[u].ch[c]==q) t[u].ch[c]=clone,u=t[u].link;
        t[q].link=t[d].link=clone;
    }
    void clear(){for(int i=1;i<=cnt;i++) e[i].clear();memset(t,0,sizeof(t));cnt=lt=1;}
    void dfs(int u){
        for(int v:e[u]) dfs(v),t[u].sz+=t[v].sz;
    }
    void build(){
        for(int i=2;i<=cnt;i++) e[t[i].link].push_back(i);
        dfs(1);
    }
    int work(){
        for(int i=2;i<=cnt;i++){
            if(t[i].sz!=k) continue;
            T.update(t[t[i].link].len+1,1);
            T.update(t[i].len+1,-1);
        }
        int maxx=1;
        for(int i=2;i<=n;i++) if(T.query(i)>=T.query(maxx)) maxx=i;
        if(T.query(maxx)==0) return -1;
        return maxx;
    }
}sam;
void solve(){
    sam.clear(),T.clear();
    read(s,k);
    n=s.size();
    for(int i:s) sam.insert(i-'a');
    sam.build();
    write(sam.work());
}
signed main(){
    int T;read(T);
    while(T--) solve(),write("\n");
    return 0;
}