题目地址(394. 字符串解码)
https://leetcode-cn.com/problems/decode-string/
题目描述
给定一个经过编码的字符串,返回它解码后的字符串。
编码规则为: 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"
前置知识
- 栈
思路
首先字符串格式比较固定,遇到k和中括号内容就需要复制k次。并且有可能嵌套在里面。不像数组顺序一样从左到右。顺序应该是先复制最里面的中括号再计算外面的中括号。也就符合栈的性质,后进先出。在想到用栈之后,那么什么时候入栈什么时候出栈呢?由于中括号内容是一个整体,那么就是遇到中括号[入栈,]出栈。知道什么时候入栈之后,我们得知道需要入栈的是什么东西,按照循环每次入栈的话,入栈的是每次遇到的单字符,而这达不到我们的要求,因为这个时候出栈的是一个元素,而如果我们遇到的是[bc]两个的话,我们希望能出栈bc两个元素。那么怎么办呢?
这个时候,我们需要一个变量来缓存我们的字符,每次遇到字符,就把字符添加到该变量中,入栈的时候,就把该变量入栈,出栈的时候,就把该变量复制k次。
关键点
- 关键是能画图,把问题给理顺。手动模拟一下:2[a3[bc]]
代码
- 语言支持:Java
Java Code:
class Solution {
public String decodeString(String s) {
Stack<Integer> multiStack = new Stack<>();
Stack<String> resStack = new Stack<>();
int multi = 0;
StringBuilder res = new StringBuilder();
for(Character c: s.toCharArray()){
if(c>='0' && c<='9'){
multi = multi*10+c-'0';
}else if(c=='['){
multiStack.push(multi);
resStack.push(res.toString());
multi = 0;
res = new StringBuilder();
}else if(c == ']'){
int currMulti = multiStack.pop();
StringBuilder temp = new StringBuilder();
for(int i = 0;i<currMulti;i++){
temp.append(res);
}
res = new StringBuilder(resStack.pop()+temp);
}else{
res.append(c);
}
}
return res.toString();
}
}
复杂度分析
令 n 为数组长度。
- 时间复杂度: O ( n ) O(n) O(n)
- 空间复杂度: O ( n ) O(n) O(n)