题目
给你一个字符串 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”
提示:
1 <= s.length <= 10^5
2 <= k <= 10^4
s 中只含有小写英文字母。
我的解答
升级版的糖果消消乐。与1047. 删除字符串中的所有相邻重复项不同的是可以自定义重复单元的个数
建两个栈s_item、s_num,s_item用于存储字符串元素,s_num表示s_item中相应位置元素的最大重复数量。比如s_item的元素是[a,c,b,c,c],那么s_num中的元素是[1,1,1,2,2]
遍历字符串s,会出现以下几种情况:
- s_item是空的,将遍历元素item压入s_item,同时将1压入s_num
- s_item非空。
- 如果item与s_item的栈顶元素不同,将遍历元素item压入s_item,同时将1压入s_num
- 如果item与s_item的栈顶元素相同,将遍历元素item压入s_item,同时s_num中压入s_num[-1]+1。
判断s_num的栈顶元素,如果等于k,则弹出s_item和s_num的k个元素。
class Solution:
def removeDuplicates(self, s: str, k: int) -> str:
s_item = []
s_num = []
for item in s:
if not s_item:# 列表为空
s_item.append(item)
s_num.append(1)
else:# 列表非空
if s_item[-1]==item:
s_item.append(item)
s_num.append(s_num[-1]+1)
if s_num[-1]==k:
for i in range(k):
s_item.pop()
s_num.pop()
else:
s_item.append(item)
s_num.append(1)
return ''.join(s_item)