洛谷 P2375 【KMP】【DP】

最新推荐文章于 2024-09-12 19:02:20 发布

willem248

最新推荐文章于 2024-09-12 19:02:20 发布

阅读量123

点赞数

分类专栏：洛谷文章标签：算法 c++ 图论

转载请注明出处，谢谢。

本文链接：https://blog.csdn.net/lxylluvio/article/details/126418822

版权

洛谷专栏收录该内容

33 篇文章 0 订阅

订阅专栏

Link

num 数组实际上就是不过半且最长的前后缀的深度，然后正常跑 kmp 即可。

#include <bits/stdc++.h>

using namespace std;

const int N = (int) 1e6 + 10, mod = (int) 1e9 + 7;
char s[N];
int nxt[N], num[N];

signed main() {
  int T;
  cin >> T;
  while (T --) {
    int ans = 1;
    num[1] = 1;
    scanf ("%s", s + 1);
    int n = strlen(s + 1);
    for (int i = 2, j = 0; i <= n; i ++) {
      while (j && s[i] != s[j + 1])
        j = nxt[j];
      if (s[i] == s[j + 1])
        j ++;
      nxt[i] = j;
      num[i] = num[j] + 1;
    }
    for (int i = 2, j = 0; i <= n; i ++) {
      while (j && s[i] != s[j + 1])
        j = nxt[j];
      if (s[i] == s[j + 1])
        j ++;
      while (j * 2 > i)
        j = nxt[j];
      ans = 1LL * ans * (num[j] + 1) % mod;
    }
    cout << ans << '\n';
  }
  return 0;
}