HDU 3336-Count the string

最新推荐文章于 2021-06-15 18:35:46 发布

xianminyan

最新推荐文章于 2021-06-15 18:35:46 发布

阅读量419

点赞数

分类专栏： kmp 文章标签： kmp

本文链接：https://blog.csdn.net/yzllz001/article/details/51789326

版权

kmp 专栏收录该内容

9 篇文章 0 订阅

订阅专栏

HDU 3336

KMP+DP

#include<cstdio>  
#include<cstring>  
using namespace std;  
#define MAX 200005  
#define mod 10007  
int next[MAX],num[MAX];  
char s[MAX];  
void get_next()  
{  
    int i=0,j=-1;  
    next[0]=-1;  
    for(; s[i];)  
        if(j==-1||s[i]==s[j])  
        {  
            ++i;  
            ++j;  
            next[i]=j;  
        }  
        else  j=next[j];  
}  
int main()  
{  
    int t,n;  
    scanf("%d",&t);  
    while(t--) 
    {  
        int summ=0;  
        memset(num,0,sizeof(num));  
        scanf("%d",&n);  
        scanf("%s",s);  
        get_next();  
        for(int i=1;i<=n;++i)  
           num[next[i]]=(num[next[i]]+1)%mod;  
        for(int i=1;i<=n;++i)  
           if(num[i])  
              summ=(summ+num[i]+1)%mod;  
            else  
              summ=(summ+1)%mod;  
        printf("%d\n",summ);  
    }  
    return 0;  
}