LeetCode刷题笔记-28.实现strStr()

C代码

int *get_prefix_table(char *pp, int psize) {
    int i = 0;
    int j;
    int *dp = NULL;

    dp = calloc(psize, sizeof(int));
    dp[0] = 0;
    for (i = 1; i < psize; i++) {
        if (pp[i] == pp[dp[i-1]]) {
            dp[i] = dp[i-1] + 1;
        } else {
            for ( j = 0; j < dp[i-1]; j++) {
                if (pp[i-j] != pp[j]) {
                    break;
                }
            }
            dp[i] = j;
        }
    }
    return dp;
}

int strStr(char * haystack, char * needle){
    int *pp = NULL;
    int i, j;

    pp = get_prefix_table(needle, strlen(needle));
    j = 0;
    for (i = 0; i < strlen(haystack); i++) {
comp:
        if(haystack[i] == needle[j]) {
            j++;
            if(j == strlen(needle)) {
                return i + 1 - strlen(needle);
            }
            continue;
        }
        if(haystack[i] != needle[j]) {
            if(j > 0) {
                j = pp[j-1];
                goto comp;
            } else {
                j = 0;
                continue;
            }
        }
    }
    return -1;
}

注意点

1. KMP算法真的要下狠心搞懂prefix table的创建以及使用；
2. 当j满足strlen(needle)时，返回的i注意已经偏移了strlen(needle)的长度，所以返回结果应该与strlen(needle)按场景计算，特别注意i的+1 -1；
3. 注意计算prefix table的动态规划子问题划分；

prefix table 动态规划法(未完待续）

法1:
$$dp(n)=\{\begin{array}{ccc}dp(n)+1,& \text{if(f(n) == f(dp[n-1])}& \text{ 最优子问题}\\ j_{0..dp[n-1]},& \text{pp[i-j] == pp[j]}& \text{子问题}\end{array}$$

子问题2，主要分析存在前缀和后缀都是字符相等重叠的情况

结果

题目