【LeetCode-栈】字符串解码

题目描述

给定一个经过编码的字符串，返回它解码后的字符串。
编码规则为: 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".

题目链接： https://leetcode-cn.com/problems/decode-string/

思路

使用栈来做。定义两个栈：numStack 用来存放数字 k，strStack 用来存放字符串。定义两个变量：nums 存放当前数字（初始化为 0），curStr 存放当前字符串（初始化为空）。

• 遍历字符串：
  • 如果当前字符是数字，则计算 k 值 num，注意，k 可能不止一位；
  • 如果当前字符时左括号 [，则将当前数字 num 压入 numStack，将当前字符串 curStr 压入 strStack，并将 num 重置为 0，将 curStr 重置为空;
  • 如果当前字符是字母，则将当前字符添加到 curStr 中（curStr += str[i]）；
  • 如果当前字符时右括号 ]，则将 numsStack、strStack 的栈顶元素分别出栈，记为 numTemp 和 strTemp。将 curStr 循环 numTemp 次，然后将 strTemp + curStr 赋值给 curStr。
• 返回 curStr。

代码如下：

class Solution {
public:
    string decodeString(string s) {
        if(s.empty()) return s;

        stack<string> strStack;
        stack<int> numStack;
        int num = 0;
        string curStr = "";
        for(int i=0; i<s.length(); i++){
            if(isdigit(s[i])) num = num*10 + s[i] - '0'; // 数字可能不止一位
            else if(s[i]=='['){
                numStack.push(num);
                strStack.push(curStr);
                num = 0;
                curStr = "";
            }else if(isalpha(s[i])){
                curStr += s[i];
            }else if(s[i]==']'){
                int numTemp = numStack.top(); numStack.pop();
                string strTemp = strStack.top(); strStack.pop();
                for(int j=0; j<numTemp; j++) strTemp += curStr;
                curStr = strTemp;
            }
        }
        return curStr;
    }
};

• 时间复杂度：O(n)
• 空间复杂度：O(n)