码迷,mamicode.com
首页 > 其他好文 > 详细

HDU 3336 扩展kmp

时间:2015-06-25 00:08:32      阅读:160      评论:0      收藏:0      [点我收藏+]

标签:

 题目大意:

找到字符串中所有和前缀字符串相同的子串的个数

 

对于这种前缀的问题,通常通过扩展kmp来解决

其实吧这是我第一次做扩展kmp的题目,原来确实看过这个概念,今天突然做到,所以这个扩展kmp的模板是做到这道题直接copy的

这里用扩展kmp很好解决问题,_next[i],表示第i位开始所能匹配到的最大公共前缀长度,比如说这个长度为4,那么说明前缀1,2,3,4都出现了一次,我们只在cnt[4]++

那么最后从n到1,逆向更新cnt[i] += cnt[i+1]即可,最后得到cnt[i]就表示前缀长度为i的时候所能得到的相同的子串的个数

 1 #include <cstdio>
 2 #include <iostream>
 3 #include <cstring>
 4 
 5 using namespace std;
 6 #define N 200005
 7 #define MOD 10007
 8 char s[N];
 9 int _next[N] , cnt[N];
10 
11 void get_next(char *T){// _next[i]: 以第i位置开始的子串 与 T的公共前缀
12      int i,length = strlen(T);
13      _next[0] = length;
14      for(i = 0;i<length-1 && T[i]==T[i+1]; i++);
15           _next[1] = i;
16           int a = 1;
17           for(int k = 2; k < length; k++){
18                   int p = a+_next[a]-1, L = _next[k-a];
19                   if( (k-1)+L >= p ){
20                        int j = (p-k+1)>0? (p-k+1) : 0;
21                        while(k+j<length && T[k+j]==T[j]) j++;// 枚举(p+1,length) 与(p-k+1,length) 区间比较
22                        _next[k] = j, a = k;
23                   }
24                   else _next[k] = L;
25          }
26 }
27 int main()
28 {
29    // freopen("a.in" , "r" , stdin);
30     int T;
31     scanf("%d" , &T);
32     while(T--)
33     {
34         int n;
35         scanf("%d%s" , &n , s);
36         get_next(s);
37       //  for(int i=0 ; i<n ; i++) cout<<_next[i]<<endl;
38         memset(cnt , 0 , sizeof(cnt));
39         for(int i=0 ; i<n ; i++) cnt[_next[i]]++;
40         for(int i=n-1 ; i>=1 ; i--) cnt[i]+=cnt[i+1];
41         int sum=0;
42         for(int i=1 ; i<=n ; i++) sum = (sum+cnt[i])%MOD;
43         printf("%d\n" , sum);
44     }
45     return 0;
46 }

 

HDU 3336 扩展kmp

标签:

原文地址:http://www.cnblogs.com/CSU3901130321/p/4598958.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!