【 bzoj 3670 】 [Noi2014]動物園 - KMP

  據說這題鬆爺現場腦補hash實現kmp?嚇尿了orz
  如果不看超過一半那個限制的話num[i] 實際上就是i 跳多少次到0,這樣直接KMP一下,完了之後跳一下就可以了。

#include <bits/stdc++.h>
#define rep(i,a,b) for (int i = a , _ = b ; i <= _ ; i ++)
#define per(i,a,b) for (int i = a , _ = b ; i >= _ ; i --)

inline int rd() {
    char c = getchar();
    while (!isdigit(c)) c = getchar() ; int x = c - '0';
    while (isdigit(c = getchar())) x = x * 10 + c - '0';
    return x;
}

const int maxn = 1000007;
const int mod = 1000000007;

int n , nxt[maxn] , num[maxn];
char st[maxn];

void input() {
    scanf("%s" , st + 1);
    n = strlen(st + 1);
}

void solve() {
    nxt[1] = 0 , num[1] = 1;
    int k = 0;
    rep (i , 2 , n) {
        while (k && st[k + 1] != st[i]) k = nxt[k];
        if (st[k + 1] == st[i]) k ++;
        nxt[i] = k , num[i] = num[k] + 1;
    }
    k = 0;
    int ans = 1;
    rep (i , 2 , n) {
        while (k && st[k + 1] != st[i]) k = nxt[k];
        if (st[k + 1] == st[i]) k ++;
        while (k > i / 2) k = nxt[k];
        ans = 1ll * ans * (num[k] + 1) % mod;
    }
    printf("%d\n" , ans);
}

int main() {
    #ifndef ONLINE_JUDGE
//      freopen("data.txt" , "r" , stdin);
    #endif
    for (int T = rd();T;T--) {
        input();
        solve();
    }
    return 0;
}
發佈了135 篇原創文章 · 獲贊 6 · 訪問量 13萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章