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 + 7;
const int MAXN = 2e5 + 10;
int nxt[MAXN], vis[MAXN], n;
char str[MAXN];

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

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