HDU—— 3336 Count the string

最新推荐文章于 2024-08-02 10:00:00 发布

原创最新推荐文章于 2024-08-02 10:00:00 发布 · 480 阅读

0 ·

CC 4.0 BY-SA版权

ACM——字符串同时被 2 个专栏收录

36 篇文章

订阅专栏

ACM——KMP

6 篇文章

订阅专栏

本文介绍了如何使用KMP算法计算一个字符串中所有前缀出现的次数之和。通过实现失配函数f[i]，我们可以找到每个前缀首次出现的位置，进而计算出所有前缀出现的总和。

题意：给定一个字符串，对于该字符串的所有前缀，找出每个前缀在字符串中出现的次数，并把次数相加得出最终的结果。

解题思路：利用KMP的失配函数f[i]，因为对于每个f[i]函数的对应值j来说，长度为j的字符串已经出现一次，长度为j-1的该字符也已经出现，故++case[j]记录第一次出现的情况,当j>0时，++case[j-1]记录j-1前缀超过2次的情况；依次类推就可以得出每个前缀在字符串中出现的总和。

Code：

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn = 200009;

int next[maxn], count[maxn],n;
char pattn[maxn] ;

void getNext ()
{
    int i = 0, j = -1 ;
    next[0] = -1 ;
    memset (count, 0, sizeof (count)) ;
    while (i < n)
    {
        if (j >= 0 && pattn[i] != pattn[j]) j = next[j] ;
        else 
        {
            ++count[i] ;   
            ++i ; ++j ; next[i] = j;
            if (j > 0) ++count[j - 1] ;
        }
    }
}

int main ()
{
    int T, sum ;
    while (~scanf ("%d", &T))
    {
        while (T--)
        {
            scanf ("%d", &n) ;
            scanf ("%s", pattn) ;
            getNext () ;
            sum = 0 ;
            for (int i = 0 ; i < n ; ++i)
                sum += count[i] ;
            printf ("%d\n", sum % 10007) ;
        }
    }
    return 0 ;
}