LeetCode-394、字符串解碼-中等

LeetCode-394、字符串解碼-中等

給定一個經過編碼的字符串，返回它解碼後的字符串。編碼規則爲: k[encoded_string]，表示其中方括號內部的 encoded_string 正好重複 k 次。注意 k 保證爲正整數。你可以認爲輸入字符串總是有效的；輸入字符串中沒有額外的空格，且輸入的方括號總是符合格式要求的。此外，你可以認爲原始數據不包含數字，所有的數字只表示重複的次數 k ，例如不會出現像 3a 或 2[4] 的輸入。

示例：

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

 

代碼：

# 使用正則表達式
class Solution:
    def decodeString(self, s: str) -> str:
        def func(m):
            return int(m.group(1)) * m.group(2)
        while '[' in s:
            s = re.sub('(\d+)\[([a-zA-Z]*)\]', func, s)
        return s

# inputStr = "hello 123 world 456"
# import re
# replacedStr = re.sub("\d+", "222", inputStr)
# 輸出 replacedStr = "hello 222 world 222"

class Solution:
    def decodeString(self, s: str) -> str:
        stack = []
        res, num = '', 0
        for c in s:
            if c.isdigit():
                num = num * 10 + int(c)
            elif c == '[':
                stack.append((res, num))
                res, num = '', 0
            elif c == "]":
                tmp = stack.pop()
                res = tmp[0] + tmp[1] * res
            else:
                res += c
        return res