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

分析

使用栈保存当前正在处理的编码字符串。
保存编码字符串的次数和前缀
遇到]符号后，把栈顶的数据取出来出来
使用string的append函数添加字符串

代码

class Solution {
public:
    string decodeString(string s) {
        string res;
        if(s.empty()) {
            return res;
        }
        
        stack<pair<int, string>> st;
        string num;
        string encode;
        for(int i = 0; i < s.length(); i++){
            if(s[i] >= '0' && s[i] <= '9') {
                num += s[i];
            } else if(s[i] == '[') {
                int n = atoi(num.c_str());
                st.push(make_pair(n, encode));
                num = "";
                encode = "";
            } else if(s[i] == ']') {
                auto t = st.top();
                int n = t.first;
                string decode = t.second;
                for(int i = 0; i < n; i++) {
                    decode.append(encode);
                }
                st.pop();
                if(st.empty()) {
                    res.append(decode);
                    encode = "";
                } else {
                    encode = decode;
                }
            } else {
                encode.append(1,s[i]);
            }
        }
        res.append(encode);
        return res;
    }
};

题目链接

https://leetcode-cn.com/problems/decode-string/description/