[leetcode] 1078。バイグラム後の発生

説明

最初と2番目の単語が与えられた場合、「最初の2番目の3番目」という形式のテキストでの出現を検討します。2番目は最初の直後にあり、3番目は2番目の直後にあります。

そのような出来事ごとに、答えに「3番目」を追加し、答えを返します。

例1：

Input: text = "alice is a good girl she is a good student", first = "a", second = "good"
Output: ["girl","student"]

例2：

Input: text = "we will we will rock you", first = "we", second = "will"
Output: ["we","rock"]

注意：

1. 1 <= text.length <= 1000
2. テキストはスペースで区切られた単語で構成され、各単語は小文字の英語で構成されます。
3. 1 <= first.length、second.length <= 10
4. 1番目と2番目は小文字の英字で構成されます。

分析

タイトルの意味は次のとおりです。指定されたテキストで、指定された1番目と2番目のバイグラムの後に表示される文字を見つけます。この質問の方法は、元のテキストをスペースで区切り、すべてのトリプルを見つけることです。トリプルの最初の2文字がキーとして使用され、3番目の文字が値として使用されるため、統計が実行されます。 、そして最後に辞書で最初と2番目の値をキーとして見つけて返します。

コード

class Solution:
    def findOcurrences(self, text: str, first: str, second: str) -> List[str]:
        text_arr=text.split()
        n=len(text_arr)
        d=defaultdict(list)
        for i in range(2,n):
            key=text_arr[i-2]+' '+text_arr[i-1]
            val=text_arr[i]
            d[key].append(val)
        return d[first+' '+second]