KMP

先上代碼,整個是完整代碼,但在代碼中間插入了一些圖解,從具體題目入手:

#include<stdio.h>
#include<string.h>
#define N 1000005
/*第一行輸入是一個數字T,表示案例的數量。每個案例包含三行。第一行是兩個數字N和M(1 <= M <= 10000,1 <= N <= 1000000)。第二行包含N個
表示[1],a [2],......,a [N]的整數。第三行包含M個整數,表示b [1],b [2],......,b [M]。所有整數都在[-1000000,1000000]的範圍內。
 產量
對於每個測試用例,應該輸出一行只包含上述K的行。如果不存在這樣的K,則輸出-1。
 示例輸入
2
13 5
1 2 1 2 3 1 2 3 1 3 2 1 2
1 2 3 1 3
13 5
1 2 1 2 3 1 2 3 1 3 2 1 2
1 2 3 2 1
 示例輸出
6
-1
 */ 爲HDU中一道使用kmp算法的題目
 
int s[N];
int p[N];
int next[N];
int m,n;
//先上next數組的求法,即找待查找的字符串的(即模式串)最大相同前綴後綴
void getnext() {
	int j=0,k=-1;//k=-1因爲p[0]前面沒有字符,所以必定沒有最長相同前綴後綴,p[k]表示前綴,p[j]表示後綴 
	next[0]=-1;
	while(j<m) {
		if(k==-1||p[j]==p[k]) {
			j++;
			k++;
                        if (p[j] != p[k])
				next[j] = k;   //說明j位置前面的子字符串最大的相同前後綴長度爲k
			else
				//因爲不能出現p[j] == p[ next[j ]],否則,利用next移動後的拿來匹配的字符還是相同的,所p[k]表示前綴,p[j]表示後綴 
	next[0]=-1;
	while(j<m) {
		if(k==-1||p[j]==p[k]) {
			j++;
			k++;
                        if (p[j] != p[k])
				next[j] = k;   //說明j位置前面的子字符串最大的相同前後綴長度爲k
			else
				//因爲不能出現p[j] == p[ next[j ]],否則,利用next移動後的拿來匹配的字符還是相同的,所
                                以當出現時需要繼續遞歸,k = next[k] = next[next[k]],下面在代碼中插入一段解釋:

				next[j] = next[k];
		} else
			k=next[k];//進一步縮小前後綴可能相等的的長度
	}
}
//利用next數組,來高效的完成字符串匹配
int kmp() {
	int i=0,j=0;//j是next數組的值
	getnext();
	while(i<n) {
		if(j==-1||s[i]==p[j]) {
			i++;
			j++;
		} else
			j=next[j];
		if(j==m)
			return i;//說明整個模式串被匹配完成
	}
	return -1;
}
int main() {
	int t;
	scanf("%d",&t);
	while(t--) {
		scanf("%d%d",&n,&m);
		for(int i=0; i<n; i++)
			scanf("%d",&s[i]);
		for(int i=0; i<m; i++)
			scanf("%d",&p[i]);
		if(kmp()==-1)
			printf("-1\n");
		else
			printf("%d\n",kmp()-m+1);
	}
	return 0;
}

引用:http://blog.csdn.net/qq_33583069/article/details/51922494

 

再貼上未優化過的next求法,對於理解優化後的有困難的朋友,可以先嚐試未優化的寫法,以下代碼仍然爲該hdu的題目:

#include<stdio.h>
#include<string.h>
#define N 1000005
/*第一行輸入是一個數字T,表示案例的數量。每個案例包含三行。第一行是兩個數字N和M(1 <= M <= 10000,1 <= N <= 1000000)。第二行包含N個表示[1],a [2],......,a [N]的整數。第三行包含M個整數,表示b [1],b [2],......,b [M]。所有整數都在[-1000000,1000000]的範圍內。


產量
對於每個測試用例,應該輸出一行只包含上述K的行。如果不存在這樣的K,則輸出-1。


示例輸入
2
13 5
1 2 1 2 3 1 2 3 1 3 2 1 2
1 2 3 1 3
13 5
1 2 1 2 3 1 2 3 1 3 2 1 2
1 2 3 2 1


示例輸出
6
-1
 */

int s[N];
int p[N];
int next[N];
int m,n;
void getnext() {
	int j=0,k=-1;
	next[0]=-1;
	while(j<m) {
		if(k==-1||p[j]==p[k]) {
			j++;
			k++;


			next[j] = k;

		} else
			k=next[k];
	}
}
int kmp() {
	int i=0,j=0;
	getnext();
	while(i<n) {
		if(j==-1||s[i]==p[j]) {
			i++;
			j++;
		} else
			j=next[j];//簡化的寫法
		if(j==m)
			return i;
	}
	return -1;
}
int main() {
	int t;
	scanf("%d",&t);
	while(t--) {
		scanf("%d%d",&n,&m);
		for(int i=0; i<n; i++)
			scanf("%d",&s[i]);
		for(int i=0; i<m; i++)
			scanf("%d",&p[i]);
		if(kmp()==-1)
			printf("-1\n");
		else
			printf("%d\n",kmp()-m+1);
	}
	return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章