Codeforces963D. Frequency of String

$n \leq 100000$的一文本串，给$m \leq 100000$个询问，每次问一小串在文本串的哪一个最短的子串里出现指定次数。注意，询问串互不相同，且总长度$\leq 100000$。

比赛时不会分析复杂度QAQ没想到这么简单

互不相同的询问串，不同的长度会只有根号个。而每个长度的出现次数都是n级别的，因此总的出现次数是$n\sqrt{n}$，只要想出$O(串长+出现次数)$的匹配算法就行了。

然后SA，SAM，AC，bitset挑个写就行了。写了AC。

//#include<iostream>
#include<cstring>
#include<cstdio>
//#include<time.h>
//#include<complex>
#include<set>
//#include<queue>
#include<algorithm>
#include<stdlib.h>
using namespace std;

#define LL long long
int qread()
{
    char c; int s=0; while ((c=getchar())<'0' || c>'9');
    do s=s*10+c-'0'; while ((c=getchar())>='0' && c<='9'); return s;
}

//Pay attention to '-' , LL and double of qread!!!!

int n,lq;
#define maxn 200011
struct Edge{int to,next;}edge[maxn<<1]; int first[maxn],le=2;
void in(int x,int y) {Edge &e=edge[le]; e.to=y; e.next=first[x]; first[x]=le++;}

set<int> ss[maxn]; int ls=0,bel[maxn];
struct AC
{
    struct Node{int ch[26],fail;}a[maxn];
    int num[maxn],size;
    AC() {size=0; memset(a[0].ch,0,sizeof(a[0].ch));}
    int insert(char *s,int id)
    {
        int now=0,m=strlen(s);
        for (int i=0;i<m;i++)
        {
            int p=s[i]-'a';
            if (!a[now].ch[p])
            {
                int x=++size;
                a[now].ch[p]=x;
            }
            now=a[now].ch[p];
        }
        num[id]=now;
        return now;
    }
    
    int que[maxn],head,tail;
    void makefail()
    {
        head=tail=0;
        for (int i=0;i<26;i++)
        {
            int u=a[0].ch[i];
            if (u) que[tail++]=u,a[u].fail=0;
        }
        while (head!=tail)
        {
            int x=que[head++];
            for (int i=0;i<26;i++)
            {
                int u=a[x].ch[i];
                if (!u) {a[x].ch[i]=a[a[x].fail].ch[i]; continue;}
                que[tail++]=u;
                a[u].fail=a[a[x].fail].ch[i];
            }
        }
    }
    void buildtree() {for (int i=1;i<=size;i++) in(a[i].fail,i);}
    
    void pei(char *s)
    {
        int n=strlen(s),now=0;
        for (int i=0;i<n;i++)
        {
            now=a[now].ch[s[i]-'a'];
            if (!bel[now]) bel[now]=++ls;
            ss[bel[now]].insert(i);
        }
    }
}ac;

int ques[maxn],kk[maxn],ll[maxn],ans[maxn];

void bing(int x)
{
    int Max=x;
    for (int i=first[x];i;i=edge[i].next)
    {
        Edge &e=edge[i];
        bing(e.to);
        if (ss[bel[e.to]].size()>ss[bel[Max]].size()) Max=e.to;
    }
    
    for (int i=first[x];i;i=edge[i].next)
    {
        Edge &e=edge[i]; if (e.to==Max) continue;
        for (auto j=ss[bel[e.to]].begin();j!=ss[bel[e.to]].end();j++) ss[bel[Max]].insert(*j);
    }
    if (Max!=x)
    {
        for (auto j=ss[bel[x]].begin();j!=ss[bel[x]].end();j++) ss[bel[Max]].insert(*j);
        bel[x]=bel[Max];
    }
    
    if (ques[x])
    {
        vector<int> p;
        for (auto j=ss[bel[x]].begin();j!=ss[bel[x]].end();j++) p.push_back(*j);
        int Ans=0x3f3f3f3f;
        for (unsigned int k=kk[ques[x]],j=k-1;j<p.size();j++) Ans=min(Ans,p[j]-p[j-k+1]);
        if (Ans==0x3f3f3f3f) ans[ques[x]]=-1;
        else ans[ques[x]]=Ans+ll[ques[x]];
    }
}

char s[maxn],t[maxn];
int main()
{
    scanf("%s",s); n=strlen(s);
    lq=qread();
    for (int i=1;i<=lq;i++)
    {
        kk[i]=qread();
        scanf("%s",t); ll[i]=strlen(t);
        ques[ac.insert(t,i)]=i;
    }
    
    ac.makefail();
    ac.buildtree();
    ac.pei(s);
    bing(0);
    for (int i=1;i<=lq;i++) printf("%d\n",ans[i]);
    return 0;
}