301 字符串匹配例题 kmp

// 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 <vector>

using namespace std;


int T;

const int N = 100010;
char S[N], P[N];
int ne[N];


void solve() {
	memset(ne, 0, sizeof ne);
	memset(S, 0, sizeof S);
	memset(P, 0, sizeof P);
	cin >> S + 1 >> P + 1 ;
	int n = strlen(P + 1);
	int m = strlen(S + 1);
	
	for (int i = 2, j = 0; i <= n; i++) {
		while (j && P[i] != P[j + 1]) j = ne[j];
		if (P[i] == P[j + 1]) j++;
		ne[i] = j;
	}
	vector<int> ans;
	for (int i = 1, j = 0; i <= m; i++) {
		while (j && S[i] != P[j + 1]) j = ne[j];
		if (S[i] == P[j + 1]) j++;
		if(j==n)
		{
			ans.push_back(i - n + 1);
			j = ne[j];
		}
	}
	if (ans.empty()) {
		cout << -1 << endl;
		cout << -1 << endl;
	}
	else {
		cout << ans.size()<< endl;
		for (int i = 0; i < ans.size(); i++) {
			cout << ans[i] << " ";
		}
		cout << endl;
	}

	return ;
}


int main()
{
	cin >> T;
	while (T--) {
		solve();
	}

	return 0;
}