1700. 增减字符串匹配

1700. 增减字符串匹配

中文English

给定只含 "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]
解释：按照"IDID",0<4,4>1,1<3,3>2。

样例 2:

输入："III"
输出：[0,1,2,3]
解释：按照"III",0<1,1<2,2<3。

注意事项

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

 

 

输入测试数据 (每行一个参数)如何理解测试数据？

class Solution:
    '''
    大致思路：
    1.初始化l = [i for i in range(len(S)+1],根据S的长度取到最大值，然后在1到最大值之间进行append，最后返回
    '''
    def diStringMatch(self, S):
        l = [i for i  in range(len(S)+1)]
        res = []
        for i in S:
            if i == 'I':
                num = l.pop(0)
                res.append(num)
            elif i == 'D':
                num = l.pop()
                res.append(num)
        res.append(l[0])
        return res