394. 字符串解码
给定一个经过编码的字符串,返回它解码后的字符串。
编码规则为: k[encoded_string],表示其中方括号内部的 encoded_string 正好重复 k 次。注意 k 保证为正整数。
你可以认为输入字符串总是有效的;输入字符串中没有额外的空格,且输入的方括号总是符合格式要求的。
此外,你可以认为原始数据不包含数字,所有的数字只表示重复的次数 k ,例如不会出现像 3a 或 2[4] 的输入。
示例:
s = "3[a]2[bc]", 返回 "aaabcbc".
s = "3[a2[c]]", 返回 "accaccacc".
s = "2[abc]3[cd]ef", 返回 "abcabccdcdcdef".
思路
借助栈来实现,两个栈,一个用来存放数字,一个用来存放字符串。数字需要做类型转换,所以用一个整型变量存储,考虑到数字可能不是一位数,需要累加来做。
class Solution {
public:
string decodeString(string s) {
stack<int> numStack;
stack<string> strStack;
string cur = "";
string result = ""; //保存展开后的字符串
int num = 0;
for(int i=0;i<s.length();i++)
{
if(s[i]>='0'&&s[i]<='9')
{
num = 10*num + s[i] - '0';
}
else if(s[i] == '[')
{
numStack.push(num);
strStack.push(cur);
num=0;
cur.clear();
}
else if(s[i]>='a' && s[i]<='z' || s[i]>='A' && s[i]<='Z')
{
cur += s[i];
}
else if(s[i] == ']')
{
int k = numStack.top();
numStack.pop();
for(int j=0;j<k;j++) //重复的过程
{
strStack.top()+=cur;
}
cur = strStack.top();
strStack.pop();
}
}
result += cur;
return result;
}
};