942. 增减字符串匹配
难度简单422
由范围 [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"
贪心
先填I的位置,然后反向填D的位置
class Solution {
public int[] diStringMatch(String s) {
int n = s.length();
int[] res = new int[n+1];
Stack<Integer> stack = new Stack<>();
int num = 0;
for(int i = 0; i < n; i++){
if(s.charAt(i) == 'I'){
res[i] = num;
num++;
}else{
stack.push(i);
}
}
res[n] = num;
num++;
while(!stack.isEmpty()){
int idx = stack.pop();
res[idx] = num;
num++;
}
return res;
}
}