DI String Match 增減字串匹配
阿新 • • 發佈:2019-01-01
給定只含 "I"
(增大)或 "D"
(減小)的字串 S
,令 N = S.length
。
返回 [0, 1, ..., N]
的任意排列 A
使得對於所有 i = 0, ..., N-1
,都有:
- 如果
S[i] == "I"
,那麼A[i] < A[i+1]
- 如果
S[i] == "D"
,那麼A[i] > A[i+1]
示例 1:
輸出:"IDID" 輸出:[0,4,1,3,2]
示例 2:
輸出:"III" 輸出:[0,1,2,3]
示例 3:
輸出:"DDI" 輸出:[3,2,0,1]
提示:
1 <= S.length <= 1000
S
只包含字元"I"
或"D"
。
思路:我們設定small表示左邊最小的數,初始化為0,big表示最大的數,初始化為S.length(),那麼只要保證遇到I時把最小的數加進去,遇到D時把最大的數加進去,那麼最後拼湊出來的陣列一定是對的。
參考程式碼:
class Solution { public: vector<int> diStringMatch(string S) { int small = 0, big = S.length(); vector<int> res; for (int i = 0; i < S.length(); i++) { if (S[i] == 'I') res.push_back(small++); else res.push_back(big--); } res.push_back(small); return res; } };