Medium
Given an encoded string, return it's 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]
.
Examples:
s = "3[a]2[bc]", return "aaabcbc".
s = "3[a2[c]]", return "accaccacc".
s = "2[abc]3[cd]ef", return "abcabccdcdcdef".
这种类型的题第一次遇到,搜了一下答案后发现这一类问题都可以用相似的套路来解答。
我们用两个stack来暂存有效信息:一个stack记录重复次数,一个stack记录字符串。遍历字符串,如果遇到数字,即是下一个“[]"里字符串需要重复输出的次数,我们便更新repeat. 注意这里的数字有可能是两位数比如12,那么我们再遇到2的时候,就需要先把之前的1变成10,再加上2得到12,所以需要先让repeat *= 10; 如果遇到的是左括号“[", 暗示着这之前我们已经遇到了表示重复次数的数字,即将开始遇到新的需要重复的字符串。这时候我们要把刚才得到的repeat压入intStack来备用, 并且要把用来表示之前的sb压入到sbStack里面,以便之后遍历的时候把新的一段字符串append到之前的字符串后面。(比如2[ab]3[ac], 遇到第二个[时,我们先得记录下之前的abab,压进stack里面以后备用)同时让sb重新new一下回归空白字符串等待被更新。同时让repeat归零(因为已经把接下来会出现的字符串需要重复的次数压入intStack里了,那么留给下一个字符串需要重复的次数就得变为0); 如果遇到右括号,说明我们要把刚刚遍历过的字符串重复需要的次数后append到之前累积的字符串里面了. 首先我们让repeat = intStack.pop(),也就是我们之前存下来的重复次数。现在需要重复的“[]"里的字符串是sb, 我们赋值给temp, 同时让sb等于之前一直积累的sb, 然后我们衔接重复需要次数的temp到sb上,来重复当前"[]"里的字符串;如果遇到的是字母,就直接append到sb后面就好了。
class Solution {
public String decodeString(String s) {
Stack<StringBuilder> sbStack = new Stack<>();
Stack<Integer> intStack = new Stack<>();
StringBuilder sb = new StringBuilder();
int repeat = 0;
for (char c : s.toCharArray()){
if (c == '['){
sbStack.push(sb);
intStack.push(repeat);
sb = new StringBuilder();
repeat = 0;
} else if (c == ']'){
StringBuilder temp = sb;
sb = sbStack.pop();
repeat = intStack.pop();
while (repeat > 0){
sb.append(temp);
repeat--;
}
} else if (c >= '0' && c <= '9'){
repeat *= 10;
repeat += c - '0';
} else {
sb.append(c);
}
}
return sb.toString();
}
}