1. 程式人生 > >Leetcode:394. 字串解碼

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

解題思路:

棧。遍歷字串,用棧記錄下s中出現的‘[’的位置。在遍歷的過程中,一旦遇到‘]’,則進行以下操作:

  1. 找到棧頂的數字,代表最近的一個'['出現的位置,讀取'['之前的一個數字data。
  2. 獲取整個[]內的字串str。
  3. 刪除data,[],以及括號內的字串,整個是連續的可統一刪除。
  4. 新字串是data個str組成的長字串,newstr。
  5. newstr插入到原先data最高位數字位置處。
  6. 當前括號處理完畢,棧頂彈出。

C++程式碼
class Solution {
public:
    #define isNUM(x) (x>='0'&&x<='9')
    string decodeString(string s) {
        stack<int> S;//記錄上一個括號'['的位置
        int pos = 0;
        while (pos <int(s.size())) {
            if (s[pos] == '[') { S.push(pos); pos++; }
            else if (s[pos] == ']') {
               //上一個'['之前讀取一個數字
                int pre = S.top();
                int data = 0, sgn = 1;
                int pos1 = pre - 1;
                int length_num = 0;
                while (pos1 >= 0 && isNUM(s[pos1])) {
                    data += sgn*(s[pos1] - '0');
                    length_num++;
                    sgn *= 10;
                    pos1--;
                }
                //獲取'[',']'之間的一個字串
                string sub = s.substr(pre + 1, pos - pre - 1);
                //新字串替換這些
                s.erase(pos1 + 1, length_num + pos - pre + 1);
                //計算新字串
                string newstr;
                while (data > 0) { newstr += sub; data--; }
                s.insert(pos1 + 1, newstr);
                pos = pos1 + 1 + int(newstr.size());
                S.pop();
            }
            else pos++;
        }
        return s;
    }
};