给定只含 “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 <= S.length <= 10000 S 只包含字符 “I” 或 “D”。
来源:力扣(LeetCode)
'''
@author: WowlNAN
'''
class Solution:
def diStringMatch(self, S: str) -> List[int]:
N=len(S)
I=0
D=N
A=[]
for W in S:
if 'I'==W:
A.append(I)
I+=1
elif 'D'==W:
A.append(D)
D-=1
if W=='I':
A.append(I)
elif W=='D':
A.append(D)
return A