算法编程(一):实现 strStr()

109 阅读2分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第1天,点击查看活动详情

写在前面

实现strStr(),这道题的难度属于简单。这说明着有一定基础知识的技术人员都可以解决此问题。

image.png

题目解读

如果你会写C语言,或者是Java;那么一定知道strstr()方法,和indexOf方法。

因为我是一名Java开发者,所以从indexOf方法的角度说明一下。

其实indexOf这个方法还是很常用的,可以判断出子字符串在父字符串中的下标位置所在,也可以判断出是否父字符串是否包含字符串。

而这道题就是要通过自己书写的算法来实现这个效果。

这显然并不难,比如通过循环父字符串每个位置的字符,逐一对子字符串进行比对,使用一个滑块的概念来解决。

代码实现

既然有了思路,我们可以顺着这个思路解决一下这个问题。

首先先判断两种特殊情况。

一:needle为空,二:haystack和needle相同。

然后,我们来遍历haystack每个字符,保证找到最小的下标索引。

通过双重循环来解决这个问题,外层循环负责找到下标索引。

内层循环复杂遍历这个下标索引是否符合题目要求。

题目并不难可以简单看一下代码。

public static void main(String[] args) {
        Solution solution = new Solution();
        System.out.println(solution.strStr("abc", "c"));
    }

    public int strStr(String haystack, String needle) {
        if(needle == null || haystack.equals(needle)){
            return 0;
        }
        for (int i = 0; i < haystack.length(); i++) {
            if(haystack.length() - i < needle.length()){
                break;
            }
            if(haystack.charAt(i) == needle.charAt(0)){
                if(needle.length() == 1){
                    return i;
                }
                int c = 1;
                for (int j = i + 1; j < haystack.length(); j++) {
                    if(haystack.charAt(j) != needle.charAt(c)){
                        break;
                    }
                    c++;
                    if(c == needle.length()){
                        return i;
                    }
                }

            }
        }
        return -1;
    }

执行结果:

显然这个方法还是挺好使的。

image.png

indexOf

public int strStr(String haystack, String needle) {
    return haystack.indexOf(needle);
}

我还运行了一下,发现效率还是挺高的。

image.png

总结

今天这道题目看着挺简单,但是很多朋友说到KMP算法,难度应该还是挺高的,不是很了解,回头学习一下。