题目大意
给定一个经过编码的字符串,返回它解码后的字符串。
编码规则为: 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".
解题思路
方法一:
准备一个字符串栈和整型栈,用来存放遍历过程中的字符串和整数。
遍历给定字符串的每个位置:
- 如果是0~9,则不断获取数字,记为int;
- 如果是字符,则不断获取字符,记为str;
- 如果是’[’,则将之前获取到的int和str分别压入对应的栈中,并重新初始化int和str;
- 如果是’]’,表示"[xxxx]“中的字符串str已经被记录,弹出整型栈顶元素k,将str重复k次,并重新压入字符串栈中(注意,只有在碰到”[“时才重新初始化int和str,因此”]“时,str就表示”[xxxx]“中的"xxxx”);
class Solution {
public:
string decodeString(string s) {
if (s.empty()){
return "";
}
stack<int> numStack;
stack<string> strStack;
int num = 0;
string cur = "";
for (int i = 0; i < s.size(); ++i){
if (s[i] <= '9' && s[i] >= '0'){
num = num * 10 + (s[i] - '0');
}
else if (s[i] == '['){
strStack.push(cur);
numStack.push(num);
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 count = numStack.top(); numStack.pop();
for (int j = 0; j < count; ++j){
strStack.top() += cur;
}
cur = strStack.top(); strStack.pop();
}
}
}
return cur;
};
方法二:
碰到非"]“元素时,直接压入栈中;
碰到”["元素时:
- 不断弹出栈顶元素,直到栈顶元素是"[",将这些元素拼成字符串;
- 将栈顶的"["弹出后,不断弹出栈顶元素,直到栈为空或者元素不是数字为止,将这些数字拼成整数(表示字符串重复的次数)。
- 将上述字符串重复K次,并重新放回栈中;
给定的原始字符串遍历完毕后,将栈中元素一一弹出,并组合成字符串返回即可。
class Solution {
public:
string decodeString(string s) {
if (s.size() == 0) return{};
stack<string> mystack;
for (int i = 0; i < s.size(); i++)
{
if (s[i] != ']')
{
string t{ s[i] };
mystack.push(t); // 如果不是右括号,则将当前字符放入栈中
}
else //是右括号,则将左右括号中的字符弹出,并跟前面的数字组成新的字符串
{
string curInStr = "";
while (mystack.top() != "[") //将中括号中的字符拼成字符串,放入curInStr中
{
curInStr = mystack.top() + curInStr;
mystack.pop();
}
mystack.pop(); //将左括号弹出
string numStr = "";
while (mystack.size() && mystack.top()[0] >= '0' && mystack.top()[0] <= '9') //将栈中的数字依次弹出
{
numStr = mystack.top() + numStr;
mystack.pop();
}
int count = stoi(numStr); //表明当前括号中的字符串重复的次数
string tmpRes = ""; //记录新字符串
for (int j = 0; j < count; j++)
tmpRes += curInStr;
mystack.push(tmpRes); //将新字符串压入栈中
}
}
string res = "";
while (mystack.size())
{
res = mystack.top() + res;
mystack.pop();
}
return res;
}
};