【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”

提示：
1 <= s.length <= 30
s 由小写英文字母、数字和方括号 '[]'组成
s 保证是一个 有效 的输入。
s 中所有整数的取值范围为 [1, 300]

使用栈

维护 两个 栈，一个保存数字，一个保存字符串。
首先，搞清楚运算顺序，假如有嵌套的括号，那一定是从最内层开始运算的，所以外层的数字和字符串放到栈里等待。
维护两个栈 不容易想到。其实，如果以后遇到多个变量需要入栈时，优先考虑使用多个栈。等写出基础版本的代码后，再去想一想，如果维护一个栈，应该如何实现。

class Solution:
    def decodeString(self, s: str) -> str:
        stack_num, stack_str = [], []
        num, res = '', ''
        for i in s:
            if i.isdigit():
                num = num + i
            elif i == '[':
                stack_num.append(int(num))
                stack_str.append(res)
                num, res = '', ''
            elif i == ']':
                res = stack_str.pop() + res*stack_num.pop()
            else:
                res += i
        return res 

时间复杂度：$O(N)$

python 小技巧

1. 判断字符串是否为数字时，可以用 str.isdigit():，而不是 if str in "0123456789":。
2. 数字的拼接操作，可以先拼接字符串，然后转为数字。或者 先把前面的数字乘以 10 再加上后面的数字。