28. 实现 strStr()

Posted 2022-05-17 潜行前行

28. 实现 strStr()
    实现 strStr() 函数。

给你两个字符串 haystack 和 needle ，请你在 haystack 字符串中找出 needle 字符串出现的第一个位置（下标从 0 开始）。如果不存在，则返回 -1 。

说明：

当 needle 是空字符串时，我们应当返回什么值呢？这是一个在面试中很好的问题。

对于本题而言，当 needle 是空字符串时我们应当返回 0 。这与 C 语言的 strstr() 以及 Java 的 indexOf() 定义相符。

示例 1：

输入：haystack = “hello”, needle = “ll”
输出：2
示例 2：

输入：haystack = “aaaaa”, needle = “bba”
输出：-1

关键点

• KMP
• 状态机转移
• 模式串回退，自己匹配自己的前缀

class Solution 
    int[][] dp;
    String pat;
    public int strStr(String haystack, String needle) 
        init(needle);
        return search(haystack);
    

    void init(String pat)
        this.pat = pat;
        int M = pat.length();
        dp = new int[M][256]; // 255 个字符
        dp[0][pat.charAt(0)] = 1;
        int X = 0;
        for(int j=1;j<M;j++)
            for(int c = 0;c<256;c++)
                dp[j][c] = dp[X][c]; // 可回退的前缀 index
            
            dp[j][pat.charAt(j)] = j+1; // 如果匹配则下一个 i
            X = dp[X][pat.charAt(j)]; // 如上一个也有前缀匹配，且匹配位置 X 和 j 字符相同 则 j 也有回退前缀 。 自己匹配自己的前缀
        
    
    int search(String haystack)
        int n = haystack.length();
        int j = 0;
        for(int i = 0;i< n;i++)
            j = dp[j][haystack.charAt(i)];
            if(j == pat.length()) return i - pat.length()+1; // 如果 j 等模式串长度则匹配成功
        
        return -1;