Codeforces Round #146 (Div. 1) C - Cyclical Quest 后缀自动机+最小循环节

#include<bits/stdc++.h>
#define LL long long
#define fi first
#define se second
#define mk make_pair
#define PII pair<int, int>
#define PLI pair<LL, int>
#define ull unsigned long long
using namespace std;

const int N = 2e6 + ;
const int inf = 0x3f3f3f3f;
const LL INF = 0x3f3f3f3f3f3f3f3f;
const int mod = 1e9 + ;
const double eps = 1e-;

int n, q, nx[N];
char s[N];

void getNext(char *s, int n) {
    int k = ;
    for(int i = ; i < n; i++) {
        while(k && s[k] != s[i]) k = nx[k-];
        if(s[k] == s[i]) k++;
        nx[i] = k;
    }
}

struct SuffixAutomaton {
    int last, cur, cnt, ch[N<<][], id[N<<], fa[N<<], dis[N<<], sz[N<<], c[N];
    SuffixAutomaton() {cur = cnt = ;}
    void init() {
        for(int i = ; i <= cnt; i++) {
            memset(ch[i], , sizeof(ch[i]));
            sz[i] = c[i] = dis[i] = fa[i] = ;
        }
        cur = cnt = ;
    }
    int extend(int p, int c) {
        cur = ++cnt; dis[cur] = dis[p]+;
        for(; p && !ch[p][c]; p = fa[p]) ch[p][c] = cur;
        if(!p) fa[cur] = ;
        else {
            int q = ch[p][c];
            if(dis[q] == dis[p]+) fa[cur] = q;
            else {
                int nt = ++cnt; dis[nt] = dis[p]+;
                memcpy(ch[nt], ch[q], sizeof(ch[q]));
                fa[nt] = fa[q]; fa[q] = fa[cur] = nt;
                for(; ch[p][c]==q; p=fa[p]) ch[p][c] = nt;
            }
        }
        sz[cur] = ;
        return cur;
    }
    void getSize(int n) {
        for(int i = ; i <= cnt; i++) c[dis[i]]++;
        for(int i = ; i <= n; i++) c[i] += c[i-];
        for(int i = cnt; i >= ; i--) id[c[dis[i]]--] = i;
        for(int i = cnt; i >= ; i--) sz[fa[id[i]]] += sz[id[i]];
    }

    void solve() {
        scanf("%s", s + );
        n = strlen(s + );
        for(int i = , last = ; i <= n; i++)
            last = extend(last, s[i]-'a');
        getSize(n);
        scanf("%d", &q);
        while(q--) {
            scanf("%s", s + );
            n = strlen(s + );
            int tar = n;
            getNext(s + , n);
            int len = n%(n-nx[n-]) ? n : (n-nx[n-]);
            for(int i = ; i < len; i++) s[++n] = s[i];
            len = ;
            LL ans = ;
            for(int i = , p = ; i <= n; i++) {
                while(p!= && !ch[p][s[i]-'a']) p = fa[p];
                if(ch[p][s[i]-'a']) {
                    len = min(dis[p], len) + ;
                    p = ch[p][s[i]-'a'];
                }
                else len = ;
                if(len >= tar) {
                    int u = p;
                    while(u !=  && dis[fa[u]] >= tar) u = fa[u];
                    if(u != ) ans += sz[u];
                }
            }
            printf("%lld\n", ans);
        }
    }
} sam;

int main() {
    sam.solve();
    return ;
}

/*
*/