942. 增减字符串匹配

题目

由范围 [0,n] 内所有整数组成的 n + 1 个整数的排列序列可以表示为长度为 n 的字符串 s ，其中:

如果 perm[i] < perm[i + 1] ，那么 s[i] == ‘I’
如果 perm[i] > perm[i + 1] ，那么 s[i] == ‘D’
给定一个字符串 s ，重构排列 perm 并返回它。如果有多个有效排列perm，则返回其中 任何一个 。

示例 1：

输入：s = “IDID”
输出：[0,4,1,3,2]

示例 2：

输入：s = “III”
输出：[0,1,2,3]

示例 3：

输入：s = “DDI”
输出：[3,2,0,1]

提示：

1 <= s.length <= 10^5
s 只包含字符 “I” 或 “D”

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

思路

挺有意思的题目，s[i]如果是’I’，那么说明s[i+1]>s[i]，贪心地将ans[i]置为当前可选数中最小的，反之如果s[i]为’D’，那么将ans[i]置为可选数中最大的。

因为每次都从可选范围两端选一个数，不会从中间取，那么维护两个最大最小变量即可。

代码

class Solution {
    public int[] diStringMatch(String s) {
        int n = s.length();
        int[] ans = new int[n + 1];
        // 可选数的最小值
        int min = 0;
        // 可选数的最大值
        int max = n;
        for (int i = 0; i < n; i++) {
            if (s.charAt(i) == 'I') {
                ans[i] = min++;
            } else {
                ans[i] = max--;
            }
        }
        ans[n] = min;
        return ans;
    }
}