增减字符串匹配
题目描述
给定只含 "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;
}
}
