LeetCode——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 <= 105
• s 只包含字符 “I” 或 “D”

答案

我的答案

class Solution {
    public int[] diStringMatch(String s) {
        int[] arr = new int[s.length()+1];
        for (int i = 0; i < arr.length; i++) {
            arr[i] = i;
        }
        for (int i = 0; i < s.length(); i++) {
            if (s.charAt(i)=='D'){
                int x = i;
                while (i < s.length()&&s.charAt(i)=='D'){
                    i++;
                }
                sort(arr,x,i);
            }
        }
        return arr;
    }

    public void sort(int[] arr,int begin,int end){
        while (begin<end){
            int temp = arr[begin];
            arr[begin] = arr[end];
            arr[end] = temp;
            begin++;
            end--;
        }
    }   
}

官方答案

方法一：贪心

思路
考虑 perm[0] 的值，根据题意：

• 如果 s[0]=‘I’，那么令 perm[0]=0，则无论 perm[1] 为何值都满足 perm[0]<perm[1]；
• 如果 s[0]=‘D’，那么令 perm[0]=n，则无论 perm[1] 为何值都满足 perm[0]>perm[1]；

确定好 perm[0] 后，剩余的 n−1 个字符和 n 个待确定的数就变成了一个和原问题相同，但规模为 n-1 的问题。因此我们可以继续按照上述方法确定 perm[1]：如果 s[1]=‘I’，那么令 perm[1] 为剩余数字中的最小数；如果 s[1]=‘D’，那么令 perm[1] 为剩余数字中的最大数。如此循环直至剩下一个数，填入 perm[n] 中。

代码实现时，由于每次都选择的是最小数和最大数，我们可以用两个变量 lo 和 hi 表示当前剩余数字中的最小数和最大数。

代码

class Solution {
    public int[] diStringMatch(String s) {
        int n = s.length(), lo = 0, hi = n;
        int[] perm = new int[n + 1];
        for (int i = 0; i < n; ++i) {
            perm[i] = s.charAt(i) == 'I' ? lo++ : hi--;
        }
        perm[n] = lo; // 最后剩下一个数，此时 lo == hi
        return perm;
    }
}

复杂度分析

• 时间复杂度：O(n)，其中 n 是字符串 s 的长度。

• 空间复杂度：O(1)，返回值不计入空间复杂度。