DS串应用--KMP算法

题目描述

学习KMP算法，给出主串和模式串，求模式串在主串的位置

输入

第一个输入t，表示有t个实例

第二行输入第1个实例的主串，第三行输入第1个实例的模式串

以此类推

输出

第一行输出第1个实例的模式串的next值

第二行输出第1个实例的匹配位置，位置从1开始计算，如果匹配成功输出位置，匹配失败输出0

以此类推

输入样例1

3
qwertyuiop
tyu
aabbccdd
ccc
aaaabababac
abac

输出样例1

-1 0 0 
5
-1 0 1 
0
-1 0 0 1 
8

#include <iostream>
#include <string>
using namespace std;

int* GetNext(string s)
{
	int j = 0, k = -1;
	int* next = new int[s.size()];
	next[0] = -1;

	while (j < s.size()-1)
	{
		if (k == -1 || s[j] == s[k])
		{
			next[++j] = ++k;
		}
		else
		{
			k = next[k];
		}
	}

	return next;
}

void KMP(string s, string p)
{
	int i = 0, j = 0;
	int* next = GetNext(p);
	while (i < s.size() && j < (int)p.size())
	{
		if (j == -1 || s[i] == p[j])
		{
			i++; j++;
		}
		else
		{
			j = next[j];
		}
	}
	if (j == p.size())
	{
		for (int i = 0; i < p.size(); i++)
		{
			cout << next[i] << " ";
		}
		cout << endl;

		cout << i - j + 1 << endl;
	}
	else
	{
		for (int i = 0; i < p.size(); i++)
		{
			cout << next[i] << " ";
		}
		cout << endl;

		cout << "0" << endl;
	}

}

int main()
{
	int t;
	cin >> t;
	while (t--)
	{
		string s, p;
		cin >> s >> p;
		KMP(s, p);
	}
	return 0;
}