leecode刷題(17)-- 實現StrStr
leecode刷題(17)-- 實現StrStr
實現StrStr
描述:
實現strStr() 函式。
給定一個 haystack 字串和一個 needle 字串,在 haystack 字串中找出 needle 字串出現的第一個位置 (從0開始)。如果不存在,則返回-1 。
示例 1:
輸入: haystack = "hello", needle = "ll" 輸出: 2
示例 2:
輸入: haystack = "aaaaa", needle = "bba" 輸出: -1
說明:
當needle
是空字串時,我們應當返回什麼值呢?這是一個在面試中很好的問題。
對於本題而言,當needle
是空字串時我們應當返回 0 。這與C語言的strstr()
以及 Java的indexOf()
定義相符。
思路:
這道題的原意是讓我們的用自己的演算法實現 strStr()函式,但我發現使用 indexOf() 是真的快啊 (捂臉)。算了,還是不用這個函數了,我們來自己實現,我想到的只有暴力遍歷兩個字串,依次比較是否相等。還有一種是KMP演算法,但是自己看的不是很懂,這裡就不列出來了。
說下過程(暴力遍歷):
首先先做判斷,如果子字串是空字串,那麼我們返回 0 。如果子字串長度大於母字串,那麼我們返回 -1。然後遍歷母字串,不需要遍歷整個母字串,我們只需要遍歷到剩下的長度和子字串相等就好了,提高運算效率。然後對於母字串的每一個字元,我們再遍歷子字串,一個一個字元對應比較,如果對應位置有不相等的,則跳出迴圈。如果一直沒有跳出迴圈,則說明比較的字元是相等的,也即子字串能和母字串位置對應,這時我們返回起始位置就可以了。
程式碼如下:
public class StrStr { public int strStr(String haystack, String needle) { if (needle.length == 0) return 0; int m = haystack.length(), n = needle.length(); if (m < n) return -1; for (int i = 0; i <= m - n; i++) { int j = 0; for (j = 0; j < n; j++) { if (haystack.charAt(i + j) != needle.charAt(j)) break; } if (j == n) return i; } return -1; } }
再說一下使用indexOf()
的方法:
程式碼如下:
class Solution { public int strStr(String haystack, String needle) { return haystack.indexOf(needle); } }
這個函式是真的又快又簡單。