[力扣c++实现] 394. 字符串解码

394. 字符串解码

难度中等956收藏分享切换为英文接收动态反馈

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

编码规则为: k[encoded_string]，表示其中方括号内部的 encoded_string 正好重复 k 次。注意 k 保证为正整数。

你可以认为输入字符串总是有效的；输入字符串中没有额外的空格，且输入的方括号总是符合格式要求的。

此外，你可以认为原始数据不包含数字，所有的数字只表示重复的次数 k ，例如不会出现像 3a 或 2[4] 的输入。

示例 1：

输入：s = "3[a]2[bc]"
输出："aaabcbc"

示例 2：

输入：s = "3[a2[c]]"
输出："accaccacc"

示例 3：

输入：s = "2[abc]3[cd]ef"
输出："abcabccdcdcdef"

示例 4：

输入：s = "abc3[cd]xyz"
输出："abccdcdcdxyz"

1. c++

/*
 * 2021-12-5 01:06
 * Hangzhou
 */


#include <iostream> 
#include <string>
#include <vector>
#include <stack>
#include <algorithm>

using namespace std;

class Solution {
public:
    string decodeString(string s) {
        int cur = 0;
        int pos = 0;
        int num = 0;

        int i = 0;
        int size = s.size();

        string ret;
        stack<char> str_stack;
        stack<pair<int,int>> num_stack;//<number,start>

        while(i < size)
        {
            if (s[i] == '[')
            {
                ++i;
                while (i < size && s[i] != ']' && !(s[i] >= '0' && s[i] <= '9'))
                {
                    str_stack.push(s[i]);
                    i++;
                    cur++;
                }
            }
            else if (s[i] >= '0' && s[i] <= '9')
            {
                num = 0;
                while (i < size && s[i]>= '0' && s[i] <= '9')
                {
                    num *= 10;
                    num += (s[i] - '0');
                    ++i;
                }

                pair<int,int> n_p(num,cur);
                num_stack.push(n_p);

            }
            else if (s[i] == ']')
            {
                ++i;
                pair<int,int> num_pos = num_stack.top();
                num_stack.pop();

                int times = num_pos.first;
                int start = num_pos.second;

                string tmp;
                int tmpcount = start;
                while (tmpcount < cur)
                {
                    tmp.push_back(str_stack.top());
                    str_stack.pop();
                    tmpcount++;
                }
 
                reverse(tmp.begin(), tmp.end());

                cur = start;
                string substring;
                for (int cnt = 0;cnt < times;++cnt)
                {
                    substring += tmp;
                }
                for (auto &chr:substring)
                {
                    cur++;
                    str_stack.push(chr);
                }
            }
            else 
            {
              str_stack.push(s[i]);
              ++i;
              ++cur;
            }
        }
        
        while (!str_stack.empty())
        {
          ret += str_stack.top();
          str_stack.pop();
        }
        reverse(ret.begin(),ret.end());
        return ret;
    }
};

int main()
{
    Solution slu;
    //string origin_string("3[a2[cfg3[eh]]]");
    string origin_string("100[leetcode]");
    cout << slu.decodeString(origin_string)<<endl;

    return 0;    
}

2.思路讲解

关键点：存储数字之后的字符的起始位置，即pair<代表次数的数字，本次重复的字符串在辅助栈中的起始位置>