394. Decode String

https://leetcode.com/problems/decode-string/description/

题目描述

 

Given an encoded string, return it's decoded string.

The encoding rule is: k[encoded_string], where the encoded_string inside the square brackets is being repeated exactly k times. Note that k is guaranteed to be a positive integer.

You may assume that the input string is always valid; No extra white spaces, square brackets are well-formed, etc.

Furthermore, you may assume that the original data does not contain any digits and that digits are only for those repeat numbers, k. For example, there won't be input like 3a or 2[4].

Examples:

s = "3[a]2[bc]", return "aaabcbc".
s = "3[a2[c]]", return "accaccacc".
s = "2[abc]3[cd]ef", return "abcabccdcdcdef".

分析

采用栈结构，存储读取到字符。利用 st, k = [欲重复的字符串，重复次数] 的结构存储读取到的内容。

1. 当读到纯数字时，依次连接存于num字符串中；
2. 当读取到 '[' 时，将重复次数 k 压入栈，之后读取到的字母字符连接到该栈顶元素的‘欲重复字符串st’部分；
3. 当读取到 ']' 时，将栈顶元素弹出，并且将解码后的字符串连接到新的栈顶元素的‘st’部分；
4. 其余情况，为字母字符，直接连接到栈顶元素的‘st’部分。

注意：栈初始化时，一定要先定义一个出现次数为1的空字符串，以便在过程中更新、并保存最终的返回结果。

stack.append(['',1])  

class Solution(object):
    def decodeString(self, s):
        """
        :type s: str
        :rtype: str
        """
        stack=[]
        stack.append(['',1])    #栈中元素的数据结构为[欲重复的字符串，重复次数]，保存结果即重复一次的字符串
        num=''  #临时保存读取到的重复次数
        for ch in s:    
            if ch.isdigit():    #保存数字字符
                num+=ch     #可能有多个连续的数字字符
            elif ch=='[':       #数字读取完毕，开始读取该数字下的重复字符串
                stack.append(['',int(num)]) #将重复次数压入栈，重复的字符内容在之后的循环中添加
                num=''          #数字清零
            elif ch==']':       #read character over
                st,k = stack.pop()  #弹出栈顶元素，赋值给 字符串/数字 对
                stack[-1][0]+=st*k  #将该括号的内容解码，连接在栈顶元素的欲重复字符串之后
            else:
                stack[-1][0]+=ch    #若是单个字母，直接连接在栈顶元素的欲重复字符串之后
        return stack[0][0]  #返回栈中最后剩下的重复一次的字符串

方法二

        #方法二：正则表达式替换字符串
        #while '[' in s:
        #    s=re.sub(r'(\\d+)\\[([a-z]*)\\]',lambda m:int(m.group(1))*m.group(2),s)