跳转至

LeetCode: 809. 情感丰富的文字

1、题目描述

有时候人们会用重复写一些字母来表示额外的感受,比如 "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 中的单词都只由小写字母组成。

2、解题思路

  • 从前向后,按照相同的字母进行分组判断
  • 如果S中的字符长度小于目标字符串中的字符长度,肯定不可以
  • 如果小于3,并且长度不一致,肯定不可以
from itertools import groupby


class Solution:
    def expressiveWords(self, S: str, words: List[str]) -> int:
        ans = 0
        temp_count = [(key, len(list(count))) for key, count in groupby(S)]

        for word in words:
            current_count = [(key, len(list(count))) for key, count in groupby(word)]
            if len(current_count) == len(temp_count):
                flag = True
                for index, (key, count) in enumerate(current_count):
                    s_key, s_count = temp_count[index]

                    if key != s_key or (s_count < 3 and s_count != count) or (s_count < count):
                        flag = False
                        break
                if flag:
                    ans += 1
        return ans