『LeetCode』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 中的单词都只由小写字母组成。

题解

【情感丰富的文字】模拟

模拟

其实就是很简单的模拟题。

遍历 words 数组，分别统计其中的单词与 s 中连续字母的个数，再判断即可

# Code language: Python
class Solution:
    def expressiveWords(self, s: str, words: List[str]) -> int:
        cnt = 0
        for w in words:
            # w -> s
            n1, n2 = len(w), len(s)
            if n1 > n2: continue
            i, j = 0, 0
            while i < n1 and j < n2:
                if w[i] != s[j]:
                    break
                alp = w[i]
                cnt1, cnt2 = 0, 0
                while i < n1 and w[i] == alp:
                    cnt1 += 1
                    i += 1
                while j < n2 and s[j] == alp:
                    cnt2 += 1
                    j += 1
                if cnt1 != cnt2 and cnt2 < 3:
                    break
                if cnt1 > cnt2:
                    break
            else:
                cnt += (i == n1 and j == n2)
        return cnt

// Code language: C++
class Solution {
public:
    int expressiveWords(string s, vector<string>& words) {
        int ans = 0;
        for (string &w: words) {
            int n1 = w.size(), n2 = s.size(), i = 0, j = 0, tag = 1;
            if (n1 > n2) continue;
            for (int cnt1 = 0, cnt2 = 0; i < n1 && j < n2; cnt1 = 0, cnt2 = 0) {
                if (w[i] != s[j]) {tag = 0; break;}
                char c = w[i];
                while (i < n1 && w[i] == c) ++cnt1, ++i;
                while (j < n2 && s[j] == c) ++cnt2, ++j;
                if (cnt1 != cnt2 && cnt2 < 3) {tag = 0; break;}
                if (cnt1 > cnt2) {tag = 0; break;}
            }
            if (i != n1 || j != n2) tag = 0;
            ans += tag;
        }
        return ans;
    }
};

时间复杂度: \(O(len(s) \sum\limits_{w \in words} len(w))\)
空间复杂度: \(O(1)\)

如果对你有帮助的话，请给我点个赞吧~

欢迎前往我的博客或 Algorithm - Github 查看更多题解

小鲸鱼的梦境

『LeetCode』809 情感丰富的文字

题目

标签

题解

模拟