騰訊2020校招第一次筆試第1題

    小Q想要給他的朋友發送一個神祕字符串，但是他發現字符串的過於長了，於是小Q
發明了一種壓縮算法對字符串中重複的部分進行了壓縮，對於字符串中連續的m個相同字
符串S將會壓縮爲[m|S](m爲一個整數且1<=m<=100)，例如字符串ABCABCABC將會被壓
縮爲[3|ABC]，現在小Q的同學收到了小Q發送過來的字符串，你能幫助他進行解壓縮麼？ 

輸入描述:
輸入第一行包含一個字符串s，代表壓縮後的字符串。
S的長度<=1000;
S僅包含大寫字母、[、]、|;
解壓後的字符串長度不超過100000;
壓縮遞歸層數不超過10層;

輸出描述:
輸出一個字符串，代表解壓後的字符串。
示例1
輸入
HG[3|B[2|CA]]F
輸出
HGBCACABCACABCACAF
說明
HG[3|B[2|CA]]F−>HG[3|BCACA]F−>HGBCACABCACABCACAF

這個題，是比較容易想到用棧來處理的，因爲涉及到括號[]內的內容先處理，這是很顯然應該用棧的。而想要把[]這一段內容寫成一般的字符串，也是不難的。這道題最難處理的地方在於，[]這一段字符串寫成一般形式以後，怎麼保存下來。如果想要存到原來的字符串，因爲[]寫成一般形式後，長度可能發生變化，因此需要移動原來的字符串，這是很不方便的；而如果保存到一個新的字符串裏面，就會出現一個問題：這段新的字符串，可能又在另一對[]裏面，這個時候需要看原來的字符串和新的字符串，這也是很難操作的。

當時寫這道題的時候，這兩種辦法都想到了，但是都無法實現，後來看了別人的解法，是把原來的字符串一個一個字符地存到棧裏面，等遇到’]‘的時候，就把字符從棧中一個一個拿出來直到遇到’[’，然後就開始把這一段[]的內容寫成一般形式（正如前面所說，這是很容易想到的）。關鍵是，寫成一般形式以後，再度寫回棧裏（這是沒那麼容易想到的）。實現的代碼如下：

#include <iostream>
#include <string>
#include <stack>
using namespace std;

int main()
{
    string str;
    while (cin >> str)
    {
        stack <char> sta;
        string ans;
        for (int i = 0; i < str.size(); ++i)
        {
            if (str[i] != ']')
            {
                sta.push(str[i]);    //沒遇到']'，就要把字符壓入棧
            }
            else
            {
                char tmp_c = sta.top();
                string tmp_s;
                while (tmp_c >= 'A' && tmp_c <= 'Z')    
                	//把[m|S]中的S提取到tmp_s
                {
                    tmp_s = tmp_c + tmp_s;
                    sta.pop();
                    tmp_c = sta.top();
                }
                
                if (!sta.empty())    //去掉'|'
                {
                    sta.pop();
                    tmp_c = sta.top();
                }
                
                int n = 0, cnt = 1;    
                while (tmp_c >= '0' && tmp_c <= '9')    
                	//把[m|S]中的m提取到n，cnt爲位數（因爲m可能不止個位）
                {
                    n += (tmp_c - '0') * cnt;
                    cnt *= 10;
                    sta.pop();
                    tmp_c = sta.top();
                }
                
                if (!sta.empty())    //去掉'['
                {
                    sta.pop();
                }
                
                while (n--)    //把tmp_s壓入棧n次
                {
                    for (auto it : tmp_s)
                    {
                        sta.push(it);
                    }
                }
                
            }
        }
        while (!sta.empty())
        {
            ans = sta.top() + ans;
            sta.pop();
        }
        cout << ans << endl;
    }
    return 0;
}

更多精彩內容，敬請期待。

2020.2.27