leetcode394:字符串解码

描述：

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

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

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

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

关键：栈

思路：

以s = "3[a]2[bc]"为例

 

遇到‘3’的时候k = 3，cur= ""

遇到‘['的时候将(3,"")入栈，k\cur设置空值 栈内[(3,"")]

遇到a的时候cur+='a'

遇到']'的时候弹出栈顶 cur = last_res("") + cur("a") * k(3) = "aaa" 栈为空

遇到2的时候k=2,此时cur = "aaa",入栈(2,"aaa"),cur="",k=0此时栈内[(2,"aaa")]

遇到b\c之后cur = "bc"，此时栈内[(2,"aaa")]

遇到']'之后弹出栈顶 cur = last_res("aaa") + cur("bc") * k(2) = "aaabcbc

 

class Solution(object):
    def decodeString(self, s):
        stack , strs ,k = [] , '' ,0
        for i in s:
            if i == '[':
                stack.append((k,strs))
                strs , k = '' , 0
            elif i==']':
                cur_mul , last_res = stack.pop()
                strs = last_res + cur_mul*strs
            elif '0'<= i <='9':
                k = 10*k +int(i)
            else:
                strs += i
        return strs