KMP算法浅浅实现

简介

不用多说，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;
    }
}