Rabin-Karp字符串匹配算法

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;
}