当前位置: 代码迷 >> 综合 >> UESTC - 1696 一道简单的字符串题
  详细解决方案

UESTC - 1696 一道简单的字符串题

热度:92   发布时间:2023-11-25 14:32:05.0
       Time Limit: 3000/1000MS (Java/Others)     Memory Limit: 65535/65535KB (Java/Others)

对于一个字符串,我们可以写出这个字符串的所有非空前缀。例如“abab”,非空前缀有:“a”,“ab”,“aba”,“abab”。对于每一个前缀,我们可以计算出他在原字符串中出现的次数。所以我们可以看到,“a”在原字符串中出现了2次;
“ab”在原字符串中出现了2次;“aba”出现了1次;“abab”出现了一次。
现在,要求您计算出给定的字符串的所有非空前缀出现的次数之和。对于“abab”,答案就是2+2+1+1=6。
由于答案可能非常大,所以要求输出答案对10007取模的值。

Input

输入由2行组成;
第一行输入一个正整数n(n<=1000000),代表字符串的长度;
第二行输入一个长度为n的字符串,只由小写字母构成。

Output

输出由一行组成;
输出一个整数,代表给定的字符串的所有非空前缀出现的次数之和对10007取模的值。

Sample input and output

Sample Input Sample Output
4
abab
6

解题思路:kmp匹配算法求得next数组然后……..详细讲解kmp算法网址(http://blog.csdn.net/buaa_shang/article/details/9907183)

代码:

#include<cstdio>
#include<cstring>
int n;
const int maxn = 1000000+100;
char s[maxn];
int next[maxn];
void get_next(){next[1]=0;int i,j=0;for(i=2;i<=n;i++){while(j>0&&s[j+1]!=s[i]) { j=next[j];} if(s[j+1]==s[i]) {j+=1;}next[i]=j;}
}
int dp[maxn];
int main(){int t,i;scanf("%d",&n);scanf("%s",s+1);get_next();dp[0]=0;int sum=0;for(i=1;i<=n;i++){dp[i]=dp[next[i]]+1;sum=(sum+dp[i])%10007;}printf("%d\n",sum);return 0;
}