leetcode394.字符串解码

leetcode394.字符串解码

题目

给定一个经过编码的字符串，返回它解码后的字符串。

编码规则为: k[encoded_string]，表示其中方括号内部的 encoded_string 正好重复 k 次。注意 k 保证为正整数。

你可以认为输入字符串总是有效的；输入字符串中没有额外的空格，且输入的方括号总是符合格式要求的。

此外，你可以认为原始数据不包含数字，所有的数字只表示重复的次数 k ，例如不会出现像 3a 或 2[4] 的输入。

示例:

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/decode-string

思路

创建一个栈，遍历字符串，当前字母不是‘ ] ’,就压入栈，若是‘ ] ’，就将最近的 ‘ [ ’ 之后的子串tmp弹出栈顶，将重复次数num也弹出栈顶，复制后将num * tmp 压入栈，直到遍历完成。

代码

class Solution:
    def decodeString(self, s):
        """
        :param s: str
        :return: str
        """
        stack = []
        cnt = 0
        for alpha in s:
            cnt += 1
            if alpha == ']':
                tmp = []
                num = ""
                while stack and stack[-1] != '[':
                    tmp = [stack.pop()] + tmp
                stack.pop()
                while stack and '0' <= stack[-1] <= '9':
                    num = stack.pop() + num
                stack += tmp * int(num)
            else:
                stack.append(alpha)

        return ''.join(stack)