【力扣】809. 情感丰富的文字-输入一组查询单词，输出其中可扩张的单词数量。

809. 情感丰富的文字

有时候人们会用重复写一些字母来表示额外的感受，比如 “hello” -> “heeellooo”, “hi” -> “hiii”。我们将相邻字母都相同的一串字符定义为相同字母组，例如：“h”, “eee”, “ll”, “ooo”。

对于一个给定的字符串 S ，如果另一个单词能够通过将一些字母组扩张从而使其和 S 相同，我们将这个单词定义为可扩张的（stretchy）。扩张操作定义如下：选择一个字母组（包含字母 c ），然后往其中添加相同的字母 c 使其长度达到 3 或以上。

例如，以 “hello” 为例，我们可以对字母组 “o” 扩张得到 “hellooo”，但是无法以同样的方法得到 “helloo” 因为字母组 “oo” 长度小于 3。此外，我们可以进行另一种扩张 “ll” -> “lllll” 以获得 “helllllooo”。如果 S = “helllllooo”，那么查询词 “hello” 是可扩张的，因为可以对它执行这两种扩张操作使得 query = “hello” -> “hellooo” -> “helllllooo” = S。

输入一组查询单词，输出其中可扩张的单词数量。

示例：
输入：
S = “heeellooo”
words = [“hello”, “hi”, “helo”]
输出：1
解释：
我们能通过扩张 “hello” 的 “e” 和 “o” 来得到 “heeellooo”。
我们不能通过扩张 “helo” 来得到 “heeellooo” 因为 “ll” 的长度小于 3 。

说明：
0 <= len(S) <= 100。
0 <= len(words) <= 100。
0 <= len(words[i]) <= 100。
S 和所有在 words 中的单词都只由小写字母组成。

代码

class Solution {
public:
    int expressiveWords(string S, vector<string>& words) {
        int res = 0;
        for (auto it : words) {
            if (match(S, it))
                res++;
        }
        return res;
    }
    bool match(string s, string w) {
        int sl = s.size();
        int wl = w.size();
        int i = 0, j = 0;
        while (i < sl && j < wl) {
            if (s[i] != w[j])return false;
            int stag = 0;//用来计数当前字符在s中的个数
            while ((i + stag) < sl && s[i + stag] == s[i])stag++;
            i += stag;//更新s中i的下标

            int wtag = 0;//用来计数当前字符在w中的个数
            while ((j + wtag) < wl && w[j + wtag] == w[j])wtag++;
            j += wtag;//更新w中j的下标
            
            if (stag == wtag || (stag >= 3 && stag > wtag)) continue;//stag>wtag是防止这种情况"aaa"["aaaa"]这结果是false
            return false;
        }
        return i == sl && j == wl;//都走到了最后说明正确。
    }
};
int main() {
    //vector<int>nums = { 1,2 };
    string s = "heeellooo";
    vector<string>words = { "hello", "hi", "helo" };
    Solution S;
    
     cout<<S.expressiveWords(s,words)<<endl;//结果为1
}