394. 字符串解码
原始题目链接:https://leetcode-cn.com/problems/decode-string/
给定一个经过编码的字符串,返回它解码后的字符串。
编码规则为: 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”
解题思路:
使用栈来完成记录遇到’[‘进栈和’]‘出栈的操作,用于还原一对括号的字符串,每次遇到’['都是一个新的进栈操作,新的统计,所以要置空倍数和当前的字符串,另外,数字的倍数那里有可能存在多位的情况,所以要有进位操作,看代码及注释。
代码实现:
class Solution:
def decodeString(self, s: str) -> str:
# 使用栈stack记录数字倍数和上一个字符串
# n表示当前的数字倍数,res是解码字符串
stack, n, res = [], 0, ""
# 遍历字符串s
for c in s:
# 遇到'[',将之前的数字倍数和遍历到的字母字符串入栈
if c == '[':
stack.append([n, res])
# 数字倍数和当前字符串重新置0和置空,用于下次遇到’[‘
n, res = 0, ""
# 遇到']',将栈顶元素弹出,即与最后遇到的'['匹配
elif c == ']':
cur_n, last_res = stack.pop()
# 就算当前的字符串res,注意这里的链接顺序:last_res + cur_n * res
res = last_res + cur_n * res
elif '0' <= c <= '9':
# 有可能是连续的数字,所以有个进位操作
n = n * 10 + int(c)
else:
res += c
return res
参考文献:
https://leetcode-cn.com/problems/decode-string/solution/decode-string-fu-zhu-zhan-fa-di-gui-fa-by-jyd/