LeetCode：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”

思路

参考：https://leetcode-cn.com/problems/decode-string/solution/decode-string-fu-zhu-zhan-fa-di-gui-fa-by-jyd/

程序

class Solution:
    def decodeString(self, s: str) -> str:
        stack = []
        multi = ""  # 用来记录倍数
        res = "" # 用来记录连续的几个字母
        for c in s:
            if '0' <= c <= '9': # 
                multi += c
            if c.isalpha():
                res += c
            if c == "[": # 遇到 [ 时，压入栈中
                stack.append((int(multi), res))
                multi = ""
                res = ""
            if c == "]":
                cur_multi, last_res = stack.pop()                
                res = last_res + cur_multi*res
        return res