LeetCode解説 100本ノック 【Day13 Find the Index of the First Occurrence in a String】

LeetCodeって？

Just a moment...

leetcode.com

LeetCodeとは、アルゴリズムやデータ構造を実践を通して学べる海外のプログラミング学習サイトです。

2022年12月現在2,000問を超える問題が登録されています。

各問題では、特定の入力をしたときに特定の返り値を出力する関数を作成することを求められます。

正誤判定は自動で行われます。

使用できる言語は、C++、C#、JAVA、Python3、Go、JSなど19言語が用意されています。

私はPython3で解いています。

Pythonではじめるアルゴリズム入門 伝統的なアルゴリズムで学ぶ定石と計算量 [ 増井 敏克 ]

問題

Just a moment...

leetcode.com

タイトル：28. Find the Index of the First Occurrence in a String

難易度：Medium

内容：

文字列 haystack と needle が与えられる。
haystack 中で、needle が出現するインデックスを返す。
見つからない場合-1を返す。

例：

Input: haystack = "sadbutsad", needle = "sad"
Output: 0
※"sad"が１文字目から現れているので１文字目のインデックスを返す。
　２回以上 needle が表れている場合、１度目のほうのインデックスを返す。

Input: haystack = "leetcode", needle = "leeto"
Output: -1

回答

find メソッドを使うと1行で終わります。

return haystack.find(needle)

せっかくなので find メソッドを実装してみます。

for i in range(len(haystack) - len(needle) + 1):
    if haystack[i : i + len(needle)] == needle:
        return i
    return -1

いずれにしろ簡単ですね。

コード

コード全文です。

class Solution:
    def strStr(self, haystack: str, needle: str) -> int:
        for i in range(len(haystack) - len(needle) + 1):
            if haystack[i : i + len(needle)] == needle:
                return i
        return -1

コードはgithubにも保管しています。

GitHub - BB-engineer/LeetCode

Contribute to BB-engineer/LeetCode development by creating an account on GitHub.

github.com

結果

いまいちLeetCodeの難易度設定が分かりません。。