Leetcode942. 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"。

 

考虑类贪心算法的想法，I的位置一定可以放当前能放的元素中最小的那个，而D的位置一定能放当前能放的元素中最大的那个

 

class Solution {
public:
    vector<int> diStringMatch(string S) {
        int len=S.length(),numi=0,numd=len;
        vector<int> res;
        for(int i=0;i<len;i++){
            if(S[i]=='I'){
                res.push_back(numi++);
            }
            else if(S[i]=='D'){
                res.push_back(numd--);
            }
        }
        res.push_back(numd);
        return res;
    }
};