394. 字符串解码

题目描述

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/decode-string

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

编码规则为: 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”.
通过次数34,322提交次数66,821

思路

一个栈放数字，一个栈放字符串，然后逐个退栈后计算，最终叠加到输出字符串中

代码

class Solution {
public:
    string decodeString(string s) {
        string res = "";
        stack<int> nums; //存放数字
        stack<string> strs; // 存放字符
        int num = 0; // 记录数字
        int len = s.size();
        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') || (s[i] >= 'A' && s[i] <= 'Z'))
                res = res + s[i];

            else if(s[i] == '[')    //将‘[’前的数字压入nums栈内， 字母字符串压入strs栈内
            {
                nums.push(num);
                num = 0;
                strs.push(res);
                res = "";
            }
            else    //遇到‘]’时，操作与之相配的‘[’之间的字符
            { 
                int times = nums.top(); // 当前字符重复次数
                nums.pop();
                for(int j = 0; j < times; j++)
                    strs.top() += res;
                
                res = strs.top();
                strs.pop();
            }
        }
        return res;
    }
};