字符串解码

[louzhedong]

习题

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

编码规则为: 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"

思路

题目中同时有数字和字母，所以我们可以使用两个栈来分别保存他们

当碰到字母，加入到当前字符串

当碰到数字，重新计算当前数字

当碰到'['，将当前数字和当前字符串分别推入栈

当碰到']'，将当前字符串扩展为数字栈顶的数字的倍数

解答

javascript

/**
 * @param {string} s
 * @return {string}
 */
var decodeString = function(s) {
    var numStack = [],
        strStack = [],
        num = 0,
        str = "";
    for (var i = 0; i < s.length; i++) {
        var current = s[i];
        if (current == '[') {
            numStack.push(num);
            strStack.push(str);
            num = 0;
            str = "";
        } else if (current == ']') {
            str = strStack.pop() + str.repeat(numStack.pop());
        } else if (isNaN(current)) { // 字母
            str += current;
        } else {  // 数字
            num = num * 10 + Number(current);
        }
    }
    return str;
};

go

func decodeString(s string) string {
    numStack := make([]int, 0)
    strStack := make([]string, 0)
    num := 0
    str := ""
    for _, char := range s {
        if char == '[' {
            numStack = append(numStack, num)
            strStack = append(strStack, str)
            num = 0
            str = ""
        } else if char == ']' {
            if (len(strStack) > 0) {
                str = strStack[len(strStack) - 1] + strings.Repeat(str, numStack[len(numStack) - 1])
                strStack = strStack[: len(strStack) - 1]
                numStack = numStack[: len(numStack) - 1]
            } else {
                str = strings.Repeat(str, numStack[len(numStack) - 1])
                numStack = numStack[: len(numStack) - 1]
            }
        } else if (char >= '0' && char <= '9') {
            num = num * 10 + int(char - '0')
        } else {
            str += string(char)
        }
    }
    return str
}