301 字符串匹配例题 exkmp

// 301 字符串匹配例题.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。
//
/*
http://oj.daimayuan.top/course/22/problem/908

给你两个字符串 a,b，字符串均由小写字母组成，
现在问你 b 在 a中出现了几次。

输入有多组数据，第一行为数据组数 T
，每组数据包含两行输入，第一行为字符串 a
，第二行为字符串 b。

对于每组输入需要输出两行，其中第一行为出现次数，
第二行为每次出现时第一个字符在 a中的下标（字符串首位的下标为 1）。如果找不到，输出两行 −1。

输入格式
第一行一个整数 T。

接下来 2T 行，每行一个字符串。

输出格式
输出 2T 行，每行若干个整数，详见题面。

样例输入
3
abababa
aba
aaaaaa
bb
abcabcab
abcab
样例输出
3
1 3 5
-1
-1
2
1 4
数据规模
对于所有数据，保证 1≤T≤10
，1≤|a|,|b|≤105
，字符串均由小写字母构成。
*/


#include <iostream>
#include <cstring>
#include <algorithm>


using namespace std;

const int N = 200010;
char s[N], p[N];
int t;
int n, m;
int z[N];


void exkmp() {
	memset(z, 0, sizeof z);
	n = strlen(s + 1);
	m = strlen(p + 1);
	p[m + 1] = '#';
	for (int i = m + 2, j = 1; j <= n; i++, j++) {
		p[i] = s[j];
	}

	int L = 1, R = 0;
	z[1] = 0;
	for (int i = 2; i <= n + m + 1; i++) {
		if (i > R) {
			z[i] = 0;
		}
		else {
			int k = i - L + 1;
			z[i] = min(z[k], R - i + 1);
		}
		while (i + z[i] <= n + m + 1 && p[z[i] + 1] == p[i + z[i]])
			++z[i];
		if (i + z[i] > R) {
			L = i; R = z[i] + i - 1;
		}
	}

	int ans = 0;
	for (int i = m + 2; i <= n + m + 1; i++)
		if (z[i] == m)
			++ans;
	if (!ans)
		cout << -1 << endl << -1 << endl;
	else {
		cout << ans << endl;
		for (int i = m + 2; i <= n + m + 2; i++) {
			if (z[i] == m)
				cout << i - m - 1 <<" ";
		}
		cout << endl;
	}

}


int main() {
	cin >> t;
	while (t--) {
		memset(s, 0, sizeof s);
		memset(p, 0, sizeof p);
		cin >> s + 1 >> p + 1;
		exkmp();
	}

	return 0;
}