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记录当前的数字，用str记录当前的
• 遇到’['时，把前面的数字和字符串放到栈中。当遇到‘]'时，取数字栈栈顶元素n, 同时n出栈，把n*str字符串放到字符串栈栈顶元素的后面，同时令str赋值为字符串栈栈顶元素，同时出栈。

代码

class Solution {
public:
    bool isNum( const char& c ) {
        return c >= '0' && c <= '9';
    }

    bool isChar( const char& c ) {
        return ( c >= 'a' && c <= 'z' ) 
            || ( c >= 'A' && c <= 'Z' );
    }

    string decodeString(string s) {
        stack<int> nStack;
        stack<string> sStack;

        int index = 0;
        int num = 0;
        string str = "";

        for ( auto& c : s ) {
            if ( isNum( c ) )
                num = num * 10 + ( c - '0' );
            else if ( isChar( c ) )
                str += c;
            else if ( c == '[' ) {
                nStack.push( num );
                sStack.push( str );
                
                num = 0;
                str = "";
            }
            else if ( c == ']' ) {
                int tempNum = nStack.top();
                nStack.pop();

                while ( tempNum-- ) 
                    sStack.top() += str;
                
                str = sStack.top();
                sStack.pop();
            }
        }

        return str;
    }
};