题目链接
给定一个经过编码的字符串,返回它解码后的字符串。
编码规则为: 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”
解题思路
思路 :栈
- 使用两个栈 stack1、stack2。stack1 用来保存左括号前已经解码的字符串,stack2 用来存储左括号前的数字。
- 用 res 存储待解码的字符串、num 存储当前数字。
- 遍历字符串。
- 如果遇到数字,则累加数字到 num。
- 如果遇到左括号,将当前待解码字符串入栈 stack1,当前数字入栈 stack2,然后将 res、nums 清空。
- 如果遇到右括号,则从 stack1 的取出待解码字符串 res,从 stack2 中取出当前数字 num,将其解码拼合成字符串赋值给 res。
- 如果遇到其他情况(遇到字母),则将当前字母加入 res 中。
- 遍历完输出解码之后的字符串 res。
解题代码
class Solution:
def decodeString(self, s: str) -> str:
stack1 = []
stack2 = []
num = 0
res = ""
for ch in s:
if ch.isdigit():
num = num * 10 + int(ch)
elif ch == '[':
stack1.append(res)
stack2.append(num)
res = ""
num = 0
elif ch == ']':
cur_res = stack1.pop()
cur_num = stack2.pop()
res = cur_res + res * cur_num
else:
res += ch
return res
参考资料:datawhalechina