[LeetCode] 394. Decode String

Given an encoded string, return its 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].

Example 1:

Input: s = "3[a]2[bc]"
Output: "aaabcbc"

Example 2:

Input: s = "3[a2[c]]"
Output: "accaccacc"

Example 3:

Input: s = "2[abc]3[cd]ef"
Output: "abcabccdcdcdef"

Example 4:

Input: s = "abc3[cd]xyz"
Output: "abccdcdcdxyz"

Constraints:

• 1 <= s.length <= 30
• s consists of lowercase English letters, digits, and square brackets '[]'.
• s is guaranteed to be a valid input.
• All the integers in s are in the range [1, 300].

字符串解码。

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

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

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/decode-string
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路是用stack。需要用到两个stack，numStack记录所有出现的数字，strStack记录所有出现的string。

当遇到字母的时候，先用一个StringBuilder res暂时记录下来；

当遇到数字的时候，就放入numStack栈保存；

当遇到左括号的时候就把已经记录下来的放入string stack，这样再遇到右括号需要结算的时候，左括号之前的部分不会丢失；

当遇到右括号的时候，就从 strStack 中pop出一个元素，和 numStack.pop() 个 res 连起来，然后作为新的res，等待下一次被塞进stack里，或者被和stack里的元素连接起来；

时间O(n)

空间O(n)

Java实现

class Solution {
    public String decodeString(String s) {
        // corner case
        if (s == null || s.length() == 0) {
            return "";
        }

        // normal case
        int len = s.length();
        Stack<Integer> numStack = new Stack<>();
        Stack<String> strStack = new Stack<>();
        StringBuilder res = new StringBuilder();
        for (int i = 0; i < len; i++) {
            char c = s.charAt(i);
            if (c == '[') {
                strStack.push(res.toString());
                res = new StringBuilder();
            } else if (c == ']') {
                int num = numStack.pop();
                StringBuilder temp = new StringBuilder(strStack.pop());
                for (int j = 0; j < num; j++) {
                    temp.append(res);
                }
                res = temp;
            } else if (Character.isDigit(c)) {
                int num = s.charAt(i) - '0';
                while (i + 1 < len && Character.isDigit(s.charAt(i + 1))) {
                    num = num * 10 + s.charAt(i + 1) - '0';
                    i++;
                }
                numStack.push(num);
            } else {
                res.append(c);
            }
        }
        return res.toString();
    }
}

JavaScript实现

/**
 * @param {string} s
 * @return {string}
 */
var decodeString = function (s) {
    // corner case
    if (s === null || s.length === 0) {
        return '';
    }

    // normal case
    let numStack = [];
    let strStack = [];
    let res = '';
    let len = s.length;
    for (let i = 0; i < len; i++) {
        let c = s.charAt(i);
        if (c === '[') {
            strStack.push(res);
            res = '';
        } else if (c == ']') {
            let num = numStack.pop();
            let temp = strStack.pop();
            for (let j = 0; j < num; j++) {
                temp += res;
            }
            res = temp;
        } else if (!isNaN(c)) {
            let num = Number(c);
            while (i + 1 < len && !isNaN(s.charAt(i + 1))) {
                num = num * 10 + Number(s.charAt(i + 1));
                i++;
            }
            numStack.push(num);
        } else {
            res += c;
        }
    }
    return res;
};

LeetCode 题目总结