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;
}
}