[LeetCode每日一题] | 1078.Bigram分词

Tips：该专栏的题解整理在https://github.com/dzw001/leetcode_notebook，有需要的童鞋自取哦~

题目链接

https://leetcode-cn.com/problems/occurrences-after-bigram/

题目描述

题目难度：简单

给出第一个词 first和第二个词 second，考虑在某些文本 text 中可能以 "first second third"形式出现的情况，其中 second 紧随 first 出现，third 紧随 second 出现。

对于每种这样的情况，将第三个词 "third"添加到答案中，并返回答案。

解题思路

简单题，首先用 .split()分词，然后遍历单词列表即可。不过有一个小发现，在线评判的时候，把变量名取的短一些似乎可以让耗时变低。

# Python3
class Solution:
    def findOcurrences(self, text: str, first: str, second: str) -> List[str]:
        res = []
        words = text.split()
        for i in range(len(words) - 2):
            if words[i] == first and words[i + 1] == second:
                res.append(words[i + 2])
        return res
 2])
        return res