leetcode_942 增减字符串匹配

102 阅读1分钟

要求

给定只含 "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]

核心代码

class Solution:
    def diStringMatch(self, s: str) -> List[int]:
        left,right = 0,len(s)
        res = []
        for char in s:
            if char == "I":
                res.append(left)
                left += 1
            else:
                res.append(right)
                right -= 1
        res.append(right)
        return res

image.png

解题思路:贪心一下,每次如果是I,就把当前最小的元素放到res里,这样下一个元素必然比当前元素大,保证了increase,同理,如果是D,就把最大的放到res里,最后记得处理最后一个元素,因为ID字符串的长度只有N,但是生成的数组长度有N+1