KMP面试题

一道例题（京东2018测试开发工程师笔试编程题）

题目描述：

给定一个字符串s，请计算输出含有连续两个s作为子串的最短字符串。注意两个s可能有重叠部分。例如，“ababa"含有两个"aba”。

输入描述：

输入包括一个字符串s，字符串长度length(1 ≤ length ≤ 50)，s中每个字符都是小写字母。

输出描述：

输出一个字符串，即含有连续两个s作为子串的最短字符串。

示例：

输入

abracadabra

输出

abracadabracada

这道题求的是一个最短的，且含有两个s作为子串的字符串

我们可以求一下 s 串的 len(s) 位置的next值，这个值表示 s 串中相等且最长的前缀和后缀。

可以直接在 s 串后面加上从 next[len(s)] + 1 到 len(s) - next[len(s)] + 1，即前缀和后缀不包含的部分。

如果不明白可以自己画一个图，很好理解。

#include <bits/stdc++.h>
using namespace std;
const int N = 1e6 + 5; 

char a[N], b[N];
int Next[N], f[N]; // Next是字符串 a 和自己匹配，f是字符串 a 和 b 匹配 
int n, m;
// 求解 Next 数组
void get_next()
{
	Next[1] = 0;
	for (int i = 2, j = 0; i <= n; i++) {
		while (j > 0 && a[i] != a[j + 1]) j = Next[j];
		if (a[i] == a[j + 1]) j++;
		Next[i] = j;
	}
}

int main(void)
{
	scanf("%s", a + 1); // a 为较短的字符串 
	n = strlen(a + 1);
	get_next();
	
	cout << a + 1 << endl;
	for (int i = Next[n] + 1; i <= n - Next[n] + 1; i++) {
		cout << a[i];
	} 
	cout << endl;
	
	return 0;
}

另一道例题

判断一棵树是否为另一棵树的子树

可以将两棵树序列化，两个字符串可以唯一表示这两颗树。进而再利用KMP判断