一、题目
有时候人们会用重复写一些字母来表示额外的感受,比如 “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。
输入一组查询单词,输出其中可扩张的单词数量。
二、思路
看到题目,两个字符串比较,首先可以想到的是双指针。对于本题,我们应该先考虑清楚不满足条件的有哪些,假设i是S的指针,j是word的指针
① S的长度小于 word 的长度
② 指针i 和 指针j 所指的字符不同
③ i所指的 相同字母的数量 大于 j 所指的字母的数量,且小于3
④ i所指的 相同字母的数量 小于 j 所指的字母的数量 (只能扩,不能缩)
(以上条件都在代码中标识)
所以我们可以根据上述条件进行题解,其他的比较好说,第三点需要计算与指针字母挨着的连续的数量,所以只需要统计这个数量,即可完成。
注意:对于S = ‘aaabbbc’ , words= [ ‘aabc’] 也是成立的
三、代码
代码如下:
1.方法一
class Solution:
def expressiveWords(self, s: str, words ) -> int:
res = 0
n = len(s)
for word in words:
cnt_s = 0
cnt_w = 0
i = 0
j = 0
while i < n and j < len(word):
#条件1#
if len(word) > n:
break
#条件2#
if s[i] != word[j]:
break
ss = s[i]
cnt_s = 0
# 计算 i所指的相同字母的个数#
while i < n and s[i] == ss:
i += 1
cnt_s += 1
# 减一是因为计数的时候多往后移了一步,要在移回来
i -= 1
w = word[j]
cnt_w = 0
while j < len(word) and word[j] == w:
j += 1
cnt_w += 1
j -= 1
#条件4 or 条件3#
if cnt_w > cnt_s or (cnt_s > cnt_w and cnt_s < 3):
break
i += 1
j += 1
if j == len(word) and i == n :
res += 1
return res