leetcode 日经贴，Cpp code -Implement strStr()

原创于 2015-05-04 13:59:08 发布 · 301 阅读

0 ·

CC 4.0 BY-SA版权

算法同时被 3 个专栏收录

183 篇文章

订阅专栏

leetcode

173 篇文章

订阅专栏

Cpp

152 篇文章

订阅专栏

本文介绍了一个字符串匹配算法的具体实现，该算法使用next数组来提高搜索效率。通过预处理模式串生成next数组，能够在主串中快速查找模式串的位置。文章详细展示了算法的工作原理及其实现过程。

Implement strStr()

class Solution {
public:
    vector<int> next;
    void calc_next(const string &pat) {
        int n = pat.size();
        next.resize(n);
        next[0] = -1;
        for (int i = 1; i < n; ++i) {
            int j = i - 1;
            while (next[j] >= 0 && pat[i] != pat[next[j] + 1]) {
                j = next[j];
            }
            next[i] = pat[i] == pat[next[j] + 1]? next[j] + 1: -1;
        }
    }
    int find(const string &s, const string &pat) {
        calc_next(pat);
        int n = s.size(), m = pat.size();
        if (n < m) {
            return -1;
        }
        int j = 0, i = 0;
        for (; i < n && j < m; ++i) {
            if (s[i] == pat[j]) {
                ++j;
            } else if (j == 0 || next[j - 1] == -1) {
                if (j > 0) {
                    --i;
                }
                j = 0;
            } else {
                --i;
                j = next[j - 1] + 1;
            }
        }
        return j < m? -1:i - m;
    }
    int strStr(string haystack, string needle) {
        if (needle.empty()) {
            return 0;
        }
        return find(haystack, needle);
    }
};