题目

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

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

示例 1：

输入：text = "alice is a good girl she is a good student", first = "a", second = "good"

输出：["girl","student"]

来源：力扣（LeetCode）

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

解答

这是一道简单题，其实主要就是知道每一种编程语言如何分割字符串，C++可以使用stringstream，Go可以直接使用strings.Split()。

Go通过一个字符串数组，将分割结果保存并根据first和second进行比较即可。

C++也可以这么做，单独建立一个vector<string>，也可以直接通过一个int数组来记录当前状态，就不用耗费string数组的大小了，但是缺点就是如果first和second相同的时候需要特殊处理。

代码如下：

C++

class Solution {
public:
    vector<string> findOcurrences(string text, string first, string second) {
        vector<string> ret;
        stringstream ss(text);
        string get;
        vector<int> dp;
        int i = 0;

        if (first == second) {
            while (getline(ss, get, ' ')) {
                if (get == first) {
                    dp.push_back(1);
                } else {
                    dp.push_back(0);
                }

                if (i > 1 && dp[i - 1] == 1 && dp[i - 2] == 1) {
                    ret.push_back(get);
                }
                i++;
            }
        } else {
            while (getline(ss, get, ' ')) {
                if (get == first) {
                    dp.push_back(1);
                } else if (get == second) {
                    dp.push_back(2);
                } else {
                    dp.push_back(0);
                }

                if ((i > 1) && (dp[i - 1] == 2) && (dp[i - 2] == 1)) {
                    ret.push_back(get);
                }
                i++;
            }
        }
        
        return ret;
    }
};

Go

func findOcurrences(text string, first string, second string) []string {
	tmp := strings.Split(text, " ")
	var ret []string

	for k, v := range tmp {
		if k < 2 {
			continue
		}

		if tmp[k-1] == second && tmp[k-2] == first {
			ret = append(ret, v)
		}
	}

	return ret
}