394. 字符串解码 c++

题目描述

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

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

解法

设置两个栈，一个栈用来存储数字，一个用来存储截取的字符串

class Solution
{
public:
    string decodeString(string s)
    {

        stack<int> nums;
        int num = 0;
        stack<string> st;
        string tmp;
        int n = s.size();
        for (int i = 0; i < n; i++)
        {
            if (s[i] >= '0' && s[i] <= '9')
            {
                num = 10 * num + s[i] - '0';
            }
            else if ((s[i] >= 'a' && s[i] <= 'z') || (s[i] >= 'A' && s[i] <= 'Z'))
            {
                tmp.push_back(s[i]);
            }
            else if (s[i] == '[')
            {
                nums.push(num);
                num = 0;
                st.push(tmp);
                tmp.clear();
            }
            else
            {
                int cnt = nums.top();
                nums.pop();
                for (int j = 0; j < cnt; j++)
                {
                    st.top() += tmp;
                }
                tmp = st.top();
                st.pop();
            }
        }

        return tmp;
    }
};