Rabin-Karp是一个不错的字符串匹配算法。wiki Rabin-Karp
理解之后,发现这种算法不仅在解决一维问题时十分有效,而且二维时也很不错。
#include <stdio.h>
#include<string.h>
#define PRIME 999999
#define max_n 4005
#define max_len 205
int len1,len2,len[max_n];
//char str1[max_len],str2[max_len]; //str1为被匹配字符串,str2为pattern
char str1[max_len],str2[max_len];
int hash(char *s,int len)
{
int i;
int key=0;
for(i=0;i<len;i++)
{
key+=s[i];
key%=PRIME;
}
return key;
}
int hs;
void rehash(char *s,int len)
{
hs+=(PRIME-s[-1]);
hs+=s[len-1];
hs%=PRIME;
}
int rabin_karp(char *s,int len_s,char *sub,int len_sub) //n,m为s,sub的结尾index
{
int hsub=hash(sub,len_sub);
int i;
for(i=0,hs=hash(s+i,len_sub);i<len_s-len_sub;i++,rehash(s+i,len_sub))
{
if(hs==hsub)
{
if(strncmp(s+i,sub,len_sub-1)==0)
{
return i;
}
}
}
return -1;//not found;
}
int main()
{
freopen("input.txt","r",stdin);
gets(str1);
gets(str2);
puts(str1);
puts(str2);
printf("%d\n",rabin_karp(str1,strlen(str1),str2,strlen(str2)));
return 0;
}