Leetcode:394. 字串解碼
阿新 • • 發佈:2018-12-05
給定一個經過編碼的字串,返回它解碼後的字串。
編碼規則為: 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中出現的‘[’的位置。在遍歷的過程中,一旦遇到‘]’,則進行以下操作:
- 找到棧頂的數字,代表最近的一個'['出現的位置,讀取'['之前的一個數字data。
- 獲取整個[]內的字串str。
- 刪除data,[],以及括號內的字串,整個是連續的可統一刪除。
- 新字串是data個str組成的長字串,newstr。
- newstr插入到原先data最高位數字位置處。
- 當前括號處理完畢,棧頂彈出。
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; } }; |