LeetCode题解之增减字符串匹配

增减字符串匹配

题目描述

给定只含 "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 <= 10000
• S 只包含字符 "I" 或 "D"。

解题思路

通过题意可知每次把可以使用的数的集合中的最小值或最大值取出，并放到当前的位置即可满足增减字符串匹配的条件，因此可以使用两个变量 min 和 max 来维护当前未使用的最小和最大的数，它们对应的区间为当前未使用的数的集合。从左向右扫描字符串，如果碰到 'I'，就取出当前最小的数，否则取出当前最大的数。

复杂度分析

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

代码实现

class Solution {
    public int[] diStringMatch(String S) {
        int min = 0;
        int max = S.length();
        int[] res = new int[max+1];
        int i = 0;
        for (; i < S.length(); i ++) {
            if (S.charAt(i) == 'I') {
                res[i] = min ++;
            } else {
                res[i] = max --;
            }
        }
        res[i] = max;
        return res;
    }
}