题目
由范围 [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’,在不知道后面字符的情况下,只能取范围(0,n)数组中目前的最小值,如果遇到‘D’,在不知道后面字符的情况下,也只能取范围(0, n)数组中的最大值。
题解
def diStringMatch(self, s: str) -> List[int]: n = len(s) # 维持有序的(0, n)的列表 temp = [i for i in range(n + 1)] res = [] for i in range(n): if s[i] == 'I': # 当前列表最小值 j = temp[0] res.append(j) temp.remove(j) else: # 当前列表最大值 j = temp[len(temp)-1] res.append(j) temp.remove(j) # 加上最后一个数 res.append(temp[0]) return res