394. 字符串解码
给定一个经过编码的字符串,返回它解码后的字符串。
编码规则为: k[encoded_string],表示其中方括号内部的 encoded_string 正好重复 k 次。注意 k 保证为正整数。
你可以认为输入字符串总是有效的;输入字符串中没有额外的空格,且输入的方括号总是符合格式要求的。
此外,你可以认为原始数据不包含数字,所有的数字只表示重复的次数 k ,例如不会出现像 3a 或 2[4] 的输入。
示例 1:
输入:s = “3[a]2[bc]”
输出:“aaabcbc”
示例 2:
输入:s = “3[a2[c]]”
输出:“accaccacc”
示例 3:
输入:s = “2[abc]3[cd]ef”
输出:“abcabccdcdcdef”
示例 4:
输入:s = “abc3[cd]xyz”
输出:“abccdcdcdxyz”
def decodeString(s: str) -> str:
stack = []
res = ''
# 当前要处理的括号对应的数字
tmp_cnt = 0
for c in s:
if c.isdigit(): # 处理数字部分
tmp_cnt = tmp_cnt * 10 + int(c)
elif c == '[': # 处理左括号
stack.append((tmp_cnt, res))
tmp_cnt, res = 0, '' # 重置为0和空,重新开始进入下一个 []
elif c == ']': # 处理右括号
# 取出括号内的字符串和数字
cur_cnt, tmp_res = stack.pop()
res = tmp_res + cur_cnt * res
else: # 处理字母部分
res += c
return res