题目是LeetCode第181场周赛的第四题,链接:最长快乐前缀。具体描述:「快乐前缀」是在原字符串中既是非空前缀也是后缀(不包括原字符串自身)的字符串。给你一个字符串 s,请你返回它的最长快乐前缀。如果不存在满足题意的前缀,则返回一个空字符串。
示例1:
输入:s = "level"
输出:"l"
解释:不包括 s 自己,一共有 4 个前缀("l", "le", "lev", "leve")和 4 个后缀("l", "el", "vel", "evel")。最长的既是前缀也是后缀的字符串是 "l" 。
示例2:
输入:s = "ababab"
输出:"abab"
解释:"abab" 是最长的既是前缀也是后缀的字符串。题目允许前后缀在原字符串中重叠。
示例3:
输入:s = "leetcodeleet"
输出:"leet"
示例4:
输入:s = "a"
输出:""
这道题最简单的方法就是暴力法,不过超时就不再说了。如果熟悉KMP算法的话,就会知道这不就是KMP算法里面的Next数组在求的东西嘛。Next数组咋求的可以参考这篇博客。
JAVA版代码如下:
class Solution {
public String longestPrefix(String s) {
char[] chars = s.toCharArray();
int L = chars.length;
int[] next = new int[L + 1];
next[0] = -1;
int k = -1, j = 0;
while (j < chars.length) {
if (k == -1 || chars[j] == chars[k]) {
next[++j] = ++k;
}
else {
k = next[k];
}
}
return next[L] == -1 ? "" : s.substring(0, next[L]);
}
}
提交结果如下:
Python版代码如下:
class Solution:
def longestPrefix(self, s: str) -> str:
N = len(s)
next = [-1 for _ in range(N + 1)]
k = -1
j = 0
while j < N:
if k == -1 or s[j] == s[k]:
k += 1
j += 1
next[j] = k
else:
k = next[k]
return s[0 : next[N]]
提交结果如下: