394. 字符串解码 - 力扣(LeetCode) (leetcode-cn.com)
思路:借助两个栈,一个numStk存放数字,一个strStk存放字母或部分串。
遍历过程中:
如果s[i]为数字,更新当前遍历得数字;
如果s[i]为字母,更新当前的部分串;
如果s[i]为左括号“[”,则将当前遍历得到的数字、当前的部分串入栈;
如果s[i]为右括号“]”,则根据numStk和strStk的两个栈顶元素,将strStk的栈顶元素合成一个新的子串,并且当前串更新为这个新的子串。
当前串是从头存在到最后的,并且最后的返回值也是它。
class Solution {
public:
string decodeString(string s) {
string ans = "";
//辅助栈
stack<int> numStk;
stack<string> strStk;
//下标
int i = 0;
//当前数字和当前串
int numTmp = 0;
string strTmp = "";
//遍历
for(int i = 0; i < s.size(); i++) {
if(isdigit(s[i])) {
numTmp = numTmp * 10 + (s[i] - '0'); //更新数字
} else if(isalpha(s[i])) {
strTmp += s[i]; //更新串
} else if(s[i] == '[') {
//入栈
numStk.push(numTmp);
strStk.push(strTmp);
//清空
numTmp = 0;
strTmp = "";
} else if(s[i] == ']') {
//更新栈顶的串,即拼接括号内容(先处理最内层)
int num = numStk.top(); numStk.pop();
while(num--) {
strStk.top() += strTmp;
}
strTmp = strStk.top();
strStk.pop();
}
}
ans = strTmp;
return ans;
}
};