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;
    }
};

猜你喜欢

转载自blog.csdn.net/qq_26410101/article/details/84669197