由范围 [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"
思路:
拿到这个题目,首先要理解题目中字符串s
与排列perm
之间的关系。由于s
中的字符I
表示当前位置的数字应小于下一个位置的数字,D
表示当前位置的数字应大于下一个位置的数字,所以可以考虑使用贪心算法,根据当前字符来决定选取当前可用数字中的最小值或最大值。
- 首先定义了字符串的长度
n
,以及可用数字的下限low
和上限high
,并创建一个用于存储排列结果的数组perm
。 - 通过循环遍历字符串
s
,当遇到I
时,选择当前下限low
作为当前位置的数字,并将low
递增,以保证后续选择的数字更大。 - 当遇到
D
时,选择当前上限high
作为当前位置的数字,并将high
递减,以保证后续选择的数字更小。 - 循环结束后,由于此时
low
和high
相等,将其赋值给perm
的最后一个位置。
时间复杂度为 O(n),因为需要遍历字符串s
一次来生成排列。
空间复杂度为 O(n),用于存储排列结果的数组perm
。
/**
* @param {string} s
* @return {number[]}
*/
var diStringMatch = function (s) {
let n = s.length,
low = 0,
high = n;
const perm = new Array(n + 1).fill(0);
for (let i = 0; i < n; ++i) {
perm[i] = s[i] === 'I' ? low++ : high--;
}
perm[n] = low; // 最后剩下一个数,此时 low == high
return perm;
};