394. 字符串解码

题目描述

给定一个经过编码的字符串，返回它解码后的字符串。

编码规则为: 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]

解答

class Solution {
public:
    string decodeString(string s) {
        // 用栈解析字符串
        string res = "";
        stack<int> nums;
        stack<string> words;
        int num = 0, len = s.size();

        // 输入：s = s = "3[a2[c]]"
        // 可模拟
        // 遇到数字和字母进行记录
        // 遇到 [ 则将前面记录的数字和字母入栈，遇到 ] 则字母和数字出栈，并叠加指定次数的字母到栈顶字母
        // num = 0
        // res = "c"
        // nums = 3
        // words = "accaccacc" 
        for(int i = 0; i < len; ++i)
        {
            if(s[i] >= '0' && s[i] <= '9') // 遇到数字
            {
                num = num * 10 + (s[i] - '0');
            }
            else if(s[i] >= 'a' && s[i] <= 'z')
            {
                res += s[i];
            }
            else if(s[i] == '[') // 表明有重复字符串
            {
                nums.push(num); // 记录重复次数
                words.push(res);
                num = 0;
                res = "";
            }
            else // s[i] == ']'
            {
                int times = nums.top();
                nums.pop();
                for(int j = 0; j < times; ++j)
                {
                    words.top() += res;
                }
                res = words.top();
                words.pop();
            }
        }
        return res;
    }
};