力扣热题100_栈_394_字符串解码

---

题目链接

394. 字符串解码

给定一个经过编码的字符串，返回它解码后的字符串。
编码规则为: 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