KMP算法

描述

给你两个字符串 S 和 T，请问你能否在 S 中找到 T。

输入

两个仅包含小写字符的字符串 S 和 T。

对于 33% 的数据，|S| <= 100，|T| <= 100。

对于 100% 的数据，|S| <= 1000000，|T| <= 1000000。

输出

如果可以找到输出 YES，否则输出 NO，然后换行。

#include<iostream>
#include<string>
using namespace std;
string s, t;
int ls, lt;//记录字符串s的长度和t的长度
int next1[1000001];
 void getNext()//next数组是从t[0~i-1]前子串的前缀后缀最大值
{
	int t1 = 0, t2;
	next1[0] = t2 = -1;//将next第一位赋值赋值为-1
	while (t1 < lt)
	{
		if (t2 == -1 || t[t1] == t[t2])
		{
			t1++;
			t2++;
			next1[t1] = t2;
		}			
		else
		{
			t2 = next1[t2];
		}
	}
}
 bool kmp()
{
	int t1 =0,t2 = 0;//从0位开始匹配
	while (t1 < ls)
	{
		if (t2 == -1 || s[t1] == t[t2]) //匹配成功，继续 
		{
			t1++;
			t2++;
		}
		else
		{
			t2 = next1[t2]; //失配 
		}
		
		if (t2 == lt)
		{
			return true;
			//t2=next1[t2];
		}
		//t2==lt时，匹配成功；t1-len2+1即为第一个字母的位置 			
	}//匹配成功后，如果要继续匹配的话则t2置为next[t2] 
	return false;
}
int main()
{
	cin >> s >> t;
	ls = s.length();
	lt = t.length();
	getNext();
	if (kmp())
	{
		cout << "YES\n";
	}
	else
	{
		cout << "NO\n";
	}
	return 0;
}