简介
不用多说,KMP算法老经典了。秋招笔试很爱考,没办法,为了饭碗,学它!
以leecode某题为例:1408. 数组中的字符串匹配
可以参考这位up主的教程:
最浅显易懂的 KMP 算法讲解
代码
class Solution {
public int[] buildNext(String p){
int n = p.length();
int[] next = new int[n]; // next[i] 表示 p[0..i] 最长共公前后缀的长度
int pre = 0; //当前缀长度
for(int i = 1; i < n; i++){
if(p.charAt(i) == p.charAt(pre)){
pre ++;
next[i] = pre;
}else{
if(pre == 0) { // 防上pre-1溢出
next[i] = 0;
}else{
pre = next[pre - 1]; // 在pre之间寻找更小的公共前后缀
i--;
}
}
}
return next;
}
public int kmp(String s, String p){
int n = s.length();
int m = p.length();
int[] next = buildNext(p);
int i = 0;
int j = 0;
while(i < n && j < m){
if(s.charAt(i) == p.charAt(j)){
i++;
j++;
}else{
if(j == 0) { // 防上j-1溢出
i++;
}else{
j = next[j-1];
}
}
}
return j == m ? i - m : -1;
}
public List<String> stringMatching(String[] words) {
int n = words.length;
HashSet<String> set = new HashSet<>();
List<String> rets = new ArrayList<>();
for(int i = 0; i < n; i++){
for(int j = 0; j < n; j++){
if(i != j && kmp(words[i], words[j]) >= 0){
set.add(words[j]);
}
}
}
rets.addAll(set);
return rets;
}
}