【模板】KMP

  1. int next[N];  
  2. char str1[M],str2[N];  
  3. //str1 长,str2 短  
  4. //len1,len2,对应str1,str2的长  
  5.   
  6. void get_next(int len2)  
  7. {  
  8.     int i = 0,j = -1;  
  9.     next[0] = -1;  
  10.     while(i<len2)  
  11.     {  
  12.         if(j == -1 || str2[i] == str2[j])  
  13.         {  
  14.             i++;  
  15.             j++;  
  16.             if(str2[i] != str2[j])  
  17.                 next[i] = j;  
  18.             else  
  19.                 next[i] = next[j];  
  20.         }  
  21.         else  
  22.             j = next[j];  
  23.     }  
  24.     //计算某字符串的周期,如aaaa是4,abcd是1  
  25.     /* 
  26.     int i = 0;j = -1; 
  27.     next[0] = -1; 
  28.     while(str2[i]) 
  29.     { 
  30.         if(j == -1 || str2[i] == str2[j]) 
  31.         { 
  32.             i++;j++; 
  33.             next[i] = j; 
  34.         } 
  35.         else 
  36.         j = next[j]; 
  37.     } 
  38.     len = strlen(str); 
  39.     i = len-j; 
  40.     if(len%i==0) 
  41.     return len/i; 
  42.     else 
  43.     return 1; 
  44.     */  
  45. }  
  46.   
  47. int kmp(int len1,int len2)  
  48. {  
  49.     int i = 0,j = 0;  
  50.     get_next(len2);  
  51.     while(i<len1)  
  52.     {  
  53.         if(j == -1 || str1[i] == str2[j])  
  54.         {  
  55.             i++;  
  56.             j++  
  57.         }  
  58.         else  
  59.             j = next[j];  
  60.         /* 
  61.                 if(j == len2)//计算str2在str1中出现多少次 
  62.                 { 
  63.         cnt++; 
  64.         j= next[j]; 
  65.                 } 
  66.                 */  
  67.     }  
  68.     //return j; //j为匹配的长度  
  69.     if(j>len2)  
  70.         return 1;//这里也可以返回i-len2来获得匹配在主串中开始的位置  
  71.     else  
  72.         return 0;  
  73. }  
  74.   
  75. //数字KMP  
  76. int a[1000005],b[10005];  
  77. int next[10005],n,m;  
  78.   
  79. void getnext()  
  80. {  
  81.     int i = 0,j = -1;  
  82.     next[0] = -1;  
  83.     while(i<m)  
  84.     {  
  85.         if(j == -1 || b[i] == b[j])  
  86.         {  
  87.             i++;  
  88.             j++;  
  89.             if(b[i] == b[j])  
  90.             next[i] = next[j];  
  91.             else  
  92.             next[i] = j;  
  93.         }  
  94.         else  
  95.         j = next[j];  
  96.     }  
  97. }  
  98.   
  99. int kmp()//返回匹配位置  
  100. {  
  101.     int i = 0,j = 0;  
  102.     while(i<n)  
  103.     {  
  104.         if(a[i] == b[j])  
  105.         {  
  106.             if(j == m-1)  
  107.             return i-j+1;  
  108.             i++;  
  109.             j++;  
  110.         }  
  111.         else  
  112.         {  
  113.             j = next[j];  
  114.             if(j == -1)  
  115.             {  
  116.                 i++;  
  117.                 j = 0;  
  118.             }  
  119.         }  
  120.     }  
  121.     return -1;  
  122. }  

你可能感兴趣的:(KMP)