LeetCode—— 942 增减字符串匹配

问题描述

给定只含 "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"。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/di-string-match
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

执行结果

代码描述

思路：左右两个指针，左边小，右边大，然后当遇到 'I' 时，说明后面要增加，所以此时应该为小值，所以先把 l 加进去，然后 r++， 遇到 'D'时，说明后面要降，此时应该为大值，所以把 r 加进去，r++.  由于S的长度，和放入的数最大值-1相等， 所以r l 会相遇，且最后压入任何一个都OK。

class Solution {
public:
    vector<int> diStringMatch(string S) {
        int N = S.length();
        vector<int> res;
        int l = 0, r = N;
        for(int i = 0; i < N; ++i)
        {
            if(S[i] == 'I')
            {
                res.push_back(l++);
            }
            else
            {
                res.push_back(r--);
            }
        }
        res.push_back(l);        // 最后压入 l or r 都OK，因为两个数都已经相等
        return res;
    }
};