开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第5天,点击查看活动详情
有时候人们会用重复写一些字母来表示额外的感受,比如 "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中的单词都只由小写字母组成。
思路
本题可以使用双指针求解。题目要求输出一组词中可扩张的单词数量,问题的核心是判断一个单词是否可以扩张成给定的字符串s。用word表示words中的任意一个单词,下面我们就实现一个算法来判断word是否可以扩张成s。定义指针i和j分别指向word和s首字母:
- 判断
word[i]是否等于s[j],如果不等说明word无法扩张成s,如果相等继续 - 定义
k = 1,如果word[i + 1] === word[i],k++,i++,继续和word下一个字符比较,直到word[i + 1]!==word[i] - 定义
l = 1,如果s[j + 1] === s[j],l++,j++,继续和s中下一个字符比较,直到s[j + 1]!==s[j] - 如果
l === k或者l > k && l >= 3,说明到目前为止是可以扩张的,i++,j++继续重复上述步骤。如果l<k或者l > k && l < 3,说明word无法扩张成s。
解题
/**
* @param {string} s
* @param {string[]} words
* @return {number}
*/
var expressiveWords = function (s, words) {
let res = 0;
for (let i = 0; i < words.length; i++) {
const word = words[i];
let m = word.length - 1;
let n = s.length - 1;
while (m >= 0 && n >= 0 && n >= m) {
if (s[n] !== word[m]) {
break;
}
let k = 1;
while (word[m - 1] === word[m]) {
k++;
m--;
}
let l = 1;
while (s[n - 1] === s[n]) {
l++;
n--;
}
if (l < k || (l > k && l < 3)) {
break;
}
m--;
n--;
}
if (m < 0 && n < 0) {
res++;
}
}
return res;
};