ACM常用模板——字符串匹配——KMP

模板：

1. #define LL long long
2. #define MS(a,b) memset(a,b,sizeof(a))
3. #define FI(a,b) fill(a,a+maxn,b)
4. #define sf(n) scanf("%d",&n)
5. #define sf2(a,b) scanf("%d%d",&a,&b)
6. #define pf(n) printf("%d\n",n)
7. #define ffr(i,n) for(i=0;i<n;i++)
8. /*
9. s:abcdabcadcabcdcab
10. p: abcdcab
11. next:-1000001
13. */
14. using namespace std;
15. void getNext(const std::string& p, std::vector& ne)
16. {
17. ne.resize(p.size());
18. ne[0] = -1;
19. int i = 0, j = -1;
20. while (i != p.size() - 1)
21. {
22. if (j == -1 || p[i] == p[j])
23. {
24. ++i;
25. ++j;
26. ne[i] = j;
27. }
28. else
29. {
30. j = ne[j];
31. }
32. }
33. }
34. int kmp(const std::string& s, const std::string& p, const int sIndex = 0)//返回下标 从0开始
35. {
36. std::vectorne(p.size());
37. getNext(p, ne);//获取next数组，保存到vector中
39. int i = sIndex, j = 0;
40. while(i != s.length() && j != p.length())
41. {
42. if (j == -1 || s[i] == p[j])
43. {
44. ++i;
45. ++j;
46. }
47. else
48. {
49. j = ne[j];
50. }
51. }
52. return j == p.length() ? i - j: -1;
53. }
54. int main()
55. {
56. string a,b;
57. while(cin>>a>>b){
58. cout<<kmp(a,b,0)<<endl;//返回匹配的下标
59. }
60. return 0;
61. }

hdu1686

模型：统计字串在母串中出现的次数（字母可重复）

1. #include
2. #include
4. int next[10005];
5. char str1[1000005],str2[10005];
6. int cnt;
8. void get_next(int len2)//生成next数组
9. {
10. int i = 0,j = -1;
11. next[0] = -1;
12. while (i<len2)
13. {
14. if(j == -1 || str2[i] == str2[j])
15. {
16. i++;
17. j++;
18. if (str2[i]!=str2[j])
19. next[i] = j;
20. else
21. next[i] = next[j];
22. }
23. else
24. j = next[j];
25. }
26. }
28. int kmp(int len1,int len2)//kmp算法
29. {
30. int i=0,j=0;
31. get_next(len2);
32. while(i<len1)
33. {
34. if(j==-1||str1[i]==str2[j])
35. {
36. ++i;
37. ++j;
38. }
39. else
40. j=next[j];
41. if(j == len2)
42. {
43. cnt++;
44. j = next[j];
45. }
46. }
47. }
49. int main()
50. {
51. int n;
52. int len1,len2;
53. scanf("%d",&n);
54. getchar();
55. while(n--)
56. {
57. gets(str2);
58. gets(str1);
59. len1 = strlen(str1);
60. len2 = strlen(str2);
61. cnt = 0;
62. kmp(len1,len2);
63. printf("%d\n",cnt);
64. }
65. return 0;
66. }

next数组应用：前缀个数

以下标i为终点重复子串的个数

aabaabaabaab

2 26 29 312 4

1. void getNext(const std::string& p, std::vector& next)
2. {
3. next.resize(p.size()+1);
4. next[0] = -1;
5. int i = 0, j = -1;
6. while (i != p.size())
7. {
8. if (j == -1 || p[i] == p[j])
9. {
10. ++i;
11. ++j;
12. next[i] = j;
13. if(i%(i-next[i])==0&&i/(i-next[i])>1){
14. printf("%d %d\n",i,i/(i-next[i]));
15. }
16. }
17. else
18. {
19. j = next[j];
20. }
21. }
22. }
23. int main()
24. {
25. int n,i,Case=1;
26. char s[maxn];
27. while(~sf(n)&&n){
28. scanf("%s",s);
29. vectornext(strlen(s)+1);
30. printf("Test case #%d\n",Case++);
31. getNext(s, next);
32. printf("\n");
33. }
34. return 0;
35. }

循环节：

1. #include
2. #include
3. using namespace std;
4. int next[1000005];
5. char s[1000005];
6. void getnext()
7. {
8. int i=0,j=-1;
9. next[0]=-1;
10. int len=strlen(s);
11. while(i<len)
12. {
13. if(s[i]==s[j]||j==-1)
14. {
15. i++;
16. j++;
17. next[i]=j;
18. }
19. else
20. j=next[j];
21. }
22. }
23. int main()
24. {
25. while(scanf("%s",s)>0)
26. {
27. if(s[0]=='.')
28. break;
29. int len=strlen(s);
30. getnext();
31. if(len%(len-next[len])==0)
32. printf("%d\n",len/(len-next[len]));
33. else
34. printf("1\n");
35. }
36. return 0;
37. }