题目
有时候人们会用重复写一些字母来表示额外的感受,比如 “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 。
提示:
1 <= s.length, words.length <= 100
1 <= words[i].length <= 100
s 和所有在 words 中的单词都只由小写字母组成。
思路
重要的是读懂题意,本身不难,双指针遍历判断即可,注意某些特殊情况,详细看下方代码
题解
class Solution: def expressiveWords(self, s: str, words: List[str]) -> int: n, ans = len(s), 0 for word in words: i, j = 0, 0 counter, nw = 1, len(word) while i < n: if j < nw and s[i] == word[j]: if counter == 2:# 如果只重复两次不符合要求 if word[j-1] != word[j-2]:#这里是为了避免hello -> helllo错误判断的情况 break counter = 1 #计数器复原 i += 1 j += 1 elif j > 0 and s[i] == word[j-1]:#处理重复扩张的情况 i += 1 counter += 1 else: #不符合要求,下一个 break if i == n and j == nw: if counter == 2:#处理某些情况重复两次就结束地情况,比如s="lee",words=["le"] if word[j-1] != word[j-2]:#处理方式同上 continue ans += 1 return ans