kuangbin专题K(next数组)

题目链接: https://vjudge.net/contest/70325#problem/K

题意: 给出一个字符串 str, 求 str 的所有前缀总共出现的次数.

思路: 先求一次 next 数组, 再用 vis[i] 存储 str[i - 1] 的所有后缀中能与 str 的前缀匹配的字符串数. 那么有 vis[i] = vis[next[i]] + 1, 后面加的那个1为字符串 str[1...i - 1].

代码:

 #include <iostream>
 #include <stdio.h>
 #include <string.h>
 using namespace std;

 const int mod = 1e4 + ;
 const int MAXN = 2e5 + ;
 int nxt[MAXN], vis[MAXN], n;
 char str[MAXN];

 void get_nxt(void){
     memset(nxt, , sizeof(nxt));
     for(int i = ; i < n; i++){
         int j = nxt[i];
         while(j && str[j] != str[i]){
             j = nxt[j];
         }
         nxt[i + ] = j + (str[i] == str[j]);
     }
 }

 int main(void){
     int t;
     scanf("%d", &t);
     while(t--){
         int sol = ;
         memset(vis, , sizeof(vis));
         scanf("%d", &n);
         scanf("%s", str);
         get_nxt();
         for(int i = ; i <= n; i++){
             vis[i] += vis[nxt[i]] + ;
             sol = (sol + vis[i]) % mod;
         }
         printf("%d\n", sol);
     }
     return ;
 }