Leetcode 1209 删除字符串中的所有相邻重复项 II

Posted 2021-09-03 一杯绿茶

给你一个字符串 s，「k 倍重复项删除操作」将会从 s 中选择 k 个相邻且相等的字母，并删除它们，使被删去的字符串的左侧和右侧连在一起。

你需要对 s 重复进行无限次这样的删除操作，直到无法继续为止。

在执行完所有删除操作后，返回最终得到的字符串。

本题答案保证唯一。

示例 1：

输入：s = "abcd", k = 2
输出："abcd"
解释：没有要删除的内容。

示例 2：

输入：s = "deeedbbcccbdaa", k = 3
输出："aa"
解释：
先删除 "eee" 和 "ccc"，得到 "ddbbbdaa"
再删除 "bbb"，得到 "dddaa"
最后删除 "ddd"，得到 "aa"

示例 3：

输入：s = "pbbcggttciiippooaais", k = 2
输出："ps"

解题思路

使用一个栈，当栈为空或者栈顶元素与当前遍历元素不相同，则直接入栈，如果栈顶元素与当前元素相同，采用了一个比较巧妙的做法，让栈顶元素与当前元素进行字符串拼接后再入栈，这样就能记录下重复元素，不需要进行额外查找，然后如果重复元素的长度达到 k-1 则 pop 之后不再入栈，以此类推。

const removeDuplicates = (s, k) => {
  let stack = [];
  for (let i=0; i<s.length; i++) {
    const char = s[i];
    const topEle = stack.pop();
    if (!topEle || topEle[0] != char) {
      stack.push(topEle);
      stack.push(char);
    } else if (topEle.length < k-1) {
      stack.push(topEle + char);
    }
  }
  return stack.join("");
};