394. 字符串解码
给定一个经过编码的字符串,返回它解码后的字符串。
编码规则为: k[encoded_string]
,表示其中方括号内部的 encoded_string 正好重复 k 次。注意 k 保证为正整数。
你可以认为输入字符串总是有效的;输入字符串中没有额外的空格,且输入的方括号总是符合格式要求的。
此外,你可以认为原始数据不包含数字,所有的数字只表示重复的次数 k ,例如不会出现像 3a
或 2[4]
的输入。
示例:
s = "3[a]2[bc]", 返回 "aaabcbc". s = "3[a2[c]]", 返回 "accaccacc". s = "2[abc]3[cd]ef", 返回 "abcabccdcdcdef".
解题思路:使用栈来记录成对的curr_string和curr_num。
遍历字符串,若字符为字母,更新curr_string;
若字符为数字,更新curr_num;
若字符为'[',则将curr_string和curr_num依次压栈;
若字符为']',则弹栈两次获得pre_string和pre_num,更新curr_string = pre_string + pre_num * curr_string。
Python3代码如下:
class Solution(object):
def decodeString(self, s):
"""
:type s: str
:rtype: str
"""
stack = []
curr_string = ''
curr_num = 0
for c in s:
if c == '[':
stack.append(curr_string)
stack.append(curr_num)
curr_string = ''
curr_num = 0
elif c == ']':
pre_num = stack.pop()
pre_string = stack.pop()
curr_string = pre_string + pre_num * curr_string
elif c.isdigit():
curr_num = 10 * curr_num + int(c)
else:
curr_string += c
return curr_string