admin管理员组

文章数量:1636954

Implement strstr()

问题

实现strstr(). 返回needle(关键字)在haystack(字符串)中第一次出现的位置,如果needle不在haystack中,则返回-1。
注:strstr()是c++中的一个函数

解决方案

O(nm) runtime, O(1) space – Brute force:

public int strStr(String haystack, String needle) {
    // 从heystack开始
    for (int i = 0; ; i++) {
        // 匹配needle的字符
        for (int j = 0; ; j++) {
            // 如果needle和j一样长,则直接返回i(当前匹配的起始位置),因为已经匹配成功了
            if (j == needle.length()) {
                return i;
            }
            // 如果i+j为当前haystack的长度,则表明已经走完heystack所有的字符,并且没有匹配成功(注意如果最后一个字符正好匹配成功,则在上面一个判断就会返回)
            if (i + j == haystack.length()) {
                return -1;
            }
            // 如果当前needle和haystack的字符相同的话(因为每次不成功匹配不成功,则i移动1位,而j又重新从0开始,所以haystack的当前位置是i+j)
            if (needle.charAt(j) != haystack.charAt(i + j)) { 
                break;
            }
}
}
}

其他思考

对于这题,有一些其他有名的算法更加有效率,如Rabin-Karp algorithm, KMP algorithm, 或者Boyer-Moore algorithm。了解这些算法可以作为一个扩展训练。
当前的brute force的实现是非常直截了当的。我们从haystack的起始位置扫描needle,并且一个个匹配所有接下去的字母。如果有其中一个字母不同,我们从haystack的接下去一个字符位置重新开始匹配。

本文标签: implementstrStr