LeetCode-Python-942. 增减字符串匹配

给定只含 "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. 1 <= S.length <= 1000
  2. S 只包含字符 "I" 或 "D"

 

思路:

贪心一下,每次如果是I,就把当前最小的元素放到res里,这样下一个元素必然比当前元素大,保证了increase,

                  同理,如果是D,就把最大的放到res里,

最后记得处理最后一个元素,因为ID字符串的长度只有N,但是生成的数组长度有N+1

class Solution(object):
    def diStringMatch(self, S):
        """
        :type S: str
        :rtype: List[int]
        """
        left = 0
        right = 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
        
        

 


版权声明:本文为qq_32424059原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。