更多精彩內(nèi)容，請(qǐng)關(guān)注【力扣簡單題】。

題目

難度：★☆☆☆☆
類型：數(shù)組

給定只含 "I"（增大）或 "D"（減?。┑淖址?S ，令 N = S.length。

返回 [0, 1, ..., N] 的任意排列 A 使得對(duì)于所有 i = 0, ..., N-1，都有：

如果 S[i] == "I"，那么 A[i] < A[i+1]
如果 S[i] == "D"，那么 A[i] > A[i+1]

提示
1 <= S.length <= 1000
S 只包含字符 "I" 或 "D"。

示例

示例 1
輸出："IDID"
輸出：[0,4,1,3,2]

示例 2
輸出："III"
輸出：[0,1,2,3]

示例 3
輸出："DDI"
輸出：[3,2,0,1]

解答

為了把[0, ......, N]中的N+1個(gè)數(shù)都使用到，我們首先構(gòu)建數(shù)組nums = [0, 1, 2, ......, N]；

每當(dāng)遇到上升趨勢(shì)時(shí)，我們彈出數(shù)組中的最小值，放在結(jié)果列表res中，這樣可以確保下一個(gè)數(shù)比當(dāng)前的數(shù)大；

每當(dāng)遇到下降趨勢(shì)時(shí)，我們彈出數(shù)組中的最大值，放在結(jié)果列表res中，這樣可以確保下一個(gè)數(shù)比當(dāng)前的數(shù)??；

最后，記得把nums中剩余的最后一個(gè)數(shù)取出來放進(jìn)結(jié)果列表末尾。

class Solution:
    def diStringMatch(self, S):
        """
        :type S: str
        :rtype: List[int]
        """
        nums = list(range(len(S)+1))
        A = []
        for c in S:
            if c == 'I':
                A.append(nums.pop(0))
            elif c == 'D':
                A.append(nums.pop())
        A.append(nums.pop())
        return A

如有疑問或建議，歡迎評(píng)論區(qū)留言~