leetcode-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"

解题思路

一开始还没有什么思路，后来看了题解想出来了。

用一个栈保存当前碰到的字符情况，如果是数字，就记录，如果是[，则把数字放到栈里面，如果是字母，则放到栈里面，如果是]，则从栈里面取出来最右边的数字右边的所有字母，然后拿最右边的数字和这些字母相乘得到重复后的字符串

代码

class Solution:
    def decodeString(self, s: str) -> str:
        stack = []
        op = 0
        for ch in s:
            if ch == ']':
                cur_str = ''
                while type(stack[-1]) == str:
                    cur_str = stack.pop() + cur_str
                stack[-1] *= cur_str
            elif ch.isdecimal():
                op = op * 10 + int(ch)
            elif ch == '[':
                stack.append(op)
                op = 0
            else:
                stack.append(ch)                
        return ''.join(stack)