leetcode394.字符串解码
题目
给定一个经过编码的字符串,返回它解码后的字符串。
编码规则为: k[encoded_string],表示其中方括号内部的 encoded_string 正好重复 k 次。注意 k 保证为正整数。
你可以认为输入字符串总是有效的;输入字符串中没有额外的空格,且输入的方括号总是符合格式要求的。
此外,你可以认为原始数据不包含数字,所有的数字只表示重复的次数 k ,例如不会出现像 3a 或 2[4] 的输入。
示例:
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/decode-string
思路
创建一个栈,遍历字符串,当前字母不是‘ ] ’,就压入栈,若是‘ ] ’,就将最近的 ‘ [ ’ 之后的子串tmp弹出栈顶,将重复次数num也弹出栈顶,复制后将num * tmp 压入栈,直到遍历完成。
代码
class Solution:
def decodeString(self, s):
"""
:param s: str
:return: str
"""
stack = []
cnt = 0
for alpha in s:
cnt += 1
if alpha == ']':
tmp = []
num = ""
while stack and stack[-1] != '[':
tmp = [stack.pop()] + tmp
stack.pop()
while stack and '0' <= stack[-1] <= '9':
num = stack.pop() + num
stack += tmp * int(num)
else:
stack.append(alpha)
return ''.join(stack)