给定一个经过编码的字符串,返回它解码后的字符串。
编码规则为: 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"
提示:
1 <= s.length <= 30
s
由小写英文字母、数字和方括号'[]'
组成s
保证是一个 有效 的输入。s
中所有整数的取值范围为[1, 300]
思路和算法
本题中可能出现括号嵌套的情况,比如 2[a2[bc]],这种情况下我们可以先转化成 2[abcbc],在转化成 abcbcabcbc。我们可以把字母、数字和括号看成是独立的 TOKEN,并用栈来维护这些 TOKEN。具体的做法是,遍历这个栈:
如果当前的字符为数位,解析出一个数字(连续的多个数位)并进栈
如果当前的字符为字母或者左括号,直接进栈
如果当前的字符为右括号,开始出栈,一直到左括号出栈,出栈序列反转后拼接成一个字符串,此时取出栈顶的数字(此时栈顶一定是数字),就是这个字符串应该出现的次数,我们根据这个次数和字符串构造出新的字符串并进栈
重复如上操作,最终将栈中的元素按照从栈底到栈顶的顺序拼接起来,就得到了答案。注意:这里可以用不定长数组来模拟栈操作,方便从栈底向栈顶遍历。
class Solution {
public:
string getDigits(string &s, int &index)
{
string ret="";
while(isdigit(s[index]))
{
ret.push_back(s[index]);
index++;
}
return ret;
}
string getstring(vector<string>& v)
{
string ret;
for(const auto &s:v)
{
ret+=s;
}
return ret;
}
string decodeString(string s) {
string result;
int index = 0;
vector<string> stk;
while(index<s.size())
{
char cur=s[index];
if(isdigit(cur))
{
string digits=getDigits(s,index);
stk.push_back(digits);
}
else if(isalpha(cur)||cur=='[')
{
stk.push_back(string(1,s[index]));
index++;
}
else{
index++;
vector<string> sub;
while(stk.back()!="[")
{
sub.push_back(stk.back());
stk.pop_back();
}
reverse(sub.begin(),sub.end());
stk.pop_back();
int reptime = stoi(stk.back());
stk.pop_back();
string child=getstring(sub);
string parent;
while(reptime)
{
parent+=child;
reptime--;
}
stk.push_back(parent);
}
}
return getstring(stk);
}
};