给定一个经过编码的字符串,返回它解码后的字符串。
编码规则为: 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".
解题思路:
栈。遍历字符串,用栈记录下s中出现的‘[’的位置。在遍历的过程中,一旦遇到‘]’,则进行以下操作:
- 找到栈顶的数字,代表最近的一个'['出现的位置,读取'['之前的一个数字data。
- 获取整个[]内的字符串str。
- 删除data,[],以及括号内的字符串,整个是连续的可统一删除。
- 新字符串是data个str组成的长字符串,newstr。
- newstr插入到原先data最高位数字位置处。
- 当前括号处理完毕,栈顶弹出。
class Solution { public: #define isNUM(x) (x>='0'&&x<='9') string decodeString(string s) { stack<int> S;//记录上一个括号'['的位置 int pos = 0; while (pos <int(s.size())) { if (s[pos] == '[') { S.push(pos); pos++; } else if (s[pos] == ']') { //上一个'['之前读取一个数字 int pre = S.top(); int data = 0, sgn = 1; int pos1 = pre - 1; int length_num = 0; while (pos1 >= 0 && isNUM(s[pos1])) { data += sgn*(s[pos1] - '0'); length_num++; sgn *= 10; pos1--; } //获取'[',']'之间的一个字符串 string sub = s.substr(pre + 1, pos - pre - 1); //新字符串替换这些 s.erase(pos1 + 1, length_num + pos - pre + 1); //计算新字符串 string newstr; while (data > 0) { newstr += sub; data--; } s.insert(pos1 + 1, newstr); pos = pos1 + 1 + int(newstr.size()); S.pop(); } else pos++; } return s; } }; |