1. 程式人生 > >DI String Match 增減字串匹配

DI String Match 增減字串匹配

給定只含 "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. 1 <= S.length <= 1000
  2. 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;
    }
};