情感丰富的文字

126 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第5天,点击查看活动详情

809. 情感丰富的文字 - 力扣(LeetCode)

有时候人们会用重复写一些字母来表示额外的感受,比如 "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。定义指针ij分别指向words首字母:

  1. 判断word[i]是否等于s[j],如果不等说明word无法扩张成s,如果相等继续
  2. 定义k = 1,如果word[i + 1] === word[i]k++i++,继续和word下一个字符比较,直到word[i + 1]!==word[i]
  3. 定义l = 1,如果s[j + 1] === s[j]l++j++,继续和s中下一个字符比较,直到s[j + 1]!==s[j]
  4. 如果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;
};