LeetCode-394. 字符串解码

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

思路

借助栈来实现，两个栈，一个用来存放数字，一个用来存放字符串。数字需要做类型转换，所以用一个整型变量存储，考虑到数字可能不是一位数，需要累加来做。

class Solution {
public:
    string decodeString(string s) {
        stack<int> numStack;
        stack<string> strStack;
        string cur = "";
        string result = ""; //保存展开后的字符串
        int num = 0;
        for(int i=0;i<s.length();i++)
        {
            if(s[i]>='0'&&s[i]<='9')
            {
                num = 10*num + s[i] - '0';
            }
            else if(s[i] == '[')
            {
                numStack.push(num);
                strStack.push(cur);
                num=0;
                cur.clear();
            }
            else if(s[i]>='a' && s[i]<='z' || s[i]>='A' && s[i]<='Z')
            {
                cur += s[i];
            }
            else if(s[i] == ']')
            {
                int k = numStack.top();
                numStack.pop();
                for(int j=0;j<k;j++) //重复的过程
                {
                    strStack.top()+=cur;
                }
                cur = strStack.top();
                strStack.pop();
            }
        }
        result += cur;
        return result;
    }
};