394. 字符串解码

394. 字符串解码(面试题打卡/中等)

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/decode-string/

题干：

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

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

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

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

提示：

• 1 <= s.length <= 30
• s 由小写英文字母、数字和方括号 '[]' 组成
• s 保证是一个 有效 的输入。
• s 中所有整数的取值范围为 [1, 300]

示例：

输入：s = "3[a]2[bc]"
输出："aaabcbc"

输入：s = "3[a2[c]]"
输出："accaccacc"

输入：s = "2[abc]3[cd]ef"
输出："abcabccdcdcdef"
    
输入：s = "abc3[cd]xyz"
输出："abccdcdcdxyz"

思路：

• 创建一个空栈，用于存储解码后的字符串。
• 遍历编码字符串的每个字符：
  • 如果当前字符是数字，将数字字符转换为整数，并将其存储为重复次数。
  • 如果当前字符是字母，直接将其加入栈中。
  • 如果当前字符是左括号 [，将当前的重复次数和空字符串入栈，并将重复次数重置为0。
  • 如果当前字符是右括号 ]，开始解码过程：
    • 从栈中弹出栈顶的字符串，直到遇到空字符串。
    • 弹出栈顶的重复次数。
    • 将弹出的字符串重复相应次数，并将结果加入栈中。
• 最后，栈中只会剩下一个字符串，即解码后的结果。将其弹出并返回。

class Solution {
    public String decodeString(String s) {
        Stack<Integer> countStack = new Stack<>(); 
        Stack<String> stringStack = new Stack<>(); 
        String currentString = "";
        int currentCount = 0;

        for (char ch : s.toCharArray()) {
            if (Character.isDigit(ch)) {
                currentCount = currentCount * 10 + (ch - '0');
            } else if (ch == '[') {
                countStack.push(currentCount);
                stringStack.push(currentString);
                currentCount = 0;
                currentString = "";
            } else if (ch == ']') {
                StringBuilder decodedString = new StringBuilder(stringStack.pop());
                int repeatedCount = countStack.pop();
                for (int i = 0; i < repeatedCount; i++) {
                    decodedString.append(currentString);
                }
                currentString = decodedString.toString();
            } else {
                currentString += ch;
            }
        }
        return currentString;
    }
}