LeetCode 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".

问题分析

遍历字符串，如果遇到的是数，则把数存入num；如果遇到的是左括号，那么将num压入数栈，将左括号前面的字符串ans压入字符串栈，然后将num清0，将ans也清空；如果遇到的是右括号，那么将数栈的栈顶数弹出记为 k，然后将当前的ans重复k次，连接到字符串栈的栈顶字符串后面，然后将字符串栈的栈顶字符串弹出，保存为ans；如果遇到的是字母，则将ans连接上该字母。当字符串遍历结束后，返回结果ans。

代码实现

class Solution {
public:
    string decodeString(string s) {
        string ans = "";
        stack<int> snum;
        stack<string> sstr;
        int num = 0;
        for(int i = 0; i < s.size(); i++){
            while(s[i] >= '0' && s[i] <= '9'){
                num = num * 10 + (s[i] - '0');
                i++;
            }
            if(s[i] == '['){
                snum.push(num);
                sstr.push(ans);
                num = 0;
                ans = "";
            }
            else if(s[i] == ']'){
                int k = snum.top();
                snum.pop();
                for(int i = 0; i < k; i++){
                    sstr.top() += ans;
                }
                ans = sstr.top();
                sstr.pop();
            }
            else{
                ans += s[i];
            }
        }
        return ans;
    }
};