题目描述:
给定一个经过编码的字符串,返回它解码后的字符串。
编码规则为: 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"
提示:
1 <= s.length <= 30
s
由小写英文字母、数字和方括号'[]'
组成s
保证是一个 有效 的输入。s
中所有整数的取值范围为[1, 300]
我的作答:
我一开始没有考虑到数字可能是十位/百位的情况,and没有考虑嵌套;
class Solution:
def decodeString(self, s: str) -> str:
# 初始化栈和结果字符串
stack = []
res = ''
num = 0
# 遍历输入字符串
for c in s:
if c.isdigit():
# 处理多位数字
num = num * 10 + int(c)
elif c == '[':
# 将数字和当前结果字符串入栈
stack.append((num, res))
num = 0
res = ''
elif c == ']':
# 出栈,得到重复次数和之前的结果字符串
prev_num, prev_res = stack.pop()
# 重复当前结果字符串并拼接
res = prev_res + prev_num * res
else:
# 普通字符,直接拼接到结果字符串
res += c
return res
参考:
class Solution:
def decodeString(self, s: str) -> str:
stack, res, multi = [], "", 0
for c in s:
if c == '[':
stack.append([multi, res])
res, multi = "", 0
elif c == ']':
cur_multi, last_res = stack.pop()
res = last_res + cur_multi * res
elif '0' <= c <= '9':
multi = multi * 10 + int(c)
else:
res += c
return res