复旦13考研机试真题(1)--字符串匹配

对于主串M和模式串P，找到P在M中出现的所有子串的第一个字符在P中的位置。P中第一个字符所在的位置为0。首行的数字表示有多少组字符串。字符长度在10⁶以内。
[输入及示例]

2
ababababa
ababa
aaa
aa

[输出及示例]

0 2 4
0 1

(相邻位置之间用一个空格隔开)
分析：这个题目我在牛客网上见到过，有人用stl中string的find函数也可以AC; 但是我查了一下find的时间复杂度应该是O(m*n), 按理来说极端测试用例应该过不了才对；这个题目我就两个解法都写一下吧，还有一种用KMP算法，时间复杂度O(m+n);

解法一：

#include<iostream>     
#include<string>
#include<vector>
using namespace std; 
 int main() { 
	 int n; 
	 scanf("%d", &n);
	 for (int i = 0; i < n; i++) {
		 string a, b;
		 int index;
		 vector<int>V;
		 cin >> a >> b;
		 index = a.find(b);
		 while (index != -1) {
			 V.push_back(index);
			 index = a.find(b, index + 1);
		 }
		 for (int i = 0; i < V.size(); i++) {
			 if (i != 0)
				 printf(" ");
			 printf("%d", V[i]);
		 }
		 printf("\n");
	 }
	 return 0;
}

解法二：

#include<iostream>     
#include<string>
#include<algorithm>
#include<vector>
using namespace std;  
int nnext[1000010];
void buildnext(const string& a) {
	fill(nnext, nnext + 1000010, 0);
	nnext[0] = -1;
	int L = a.length();
	int i = 0, j = -1;
	while (i < L) {
		if (j == -1 || a[i] == a[j]) {
			i++;
			j++;
			nnext[i] = j;
		}
		else
			j = nnext[j];
	}
} 
 int main() { 
	 int n; 
	 scanf("%d", &n);
	 for (int i = 0; i < n; i++) {
		 string a, b; 
		 vector<int>V;
		 cin >> a >> b;
		 buildnext(b);
		 int La = a.length();
		 int Lb = b.length();
		 int k = -1;
		 for (int j = -1; j < La;) {
			 if (k == -1 || a[j] == b[k]) {
				 j++;
				 k++;
			 }
			 else
				 k = nnext[k];
			 if (k == Lb) {
				 V.push_back(j - Lb);
				 k = nnext[k];
			 }  
		 }
		 for (int i = 0; i < V.size(); i++) {
			 if (i != 0)
				 printf(" ");
			 printf("%d", V[i]);
		 }
		 printf("\n");
	 }
	 return 0;
}