跳转至

LeetCode: 942. 增减字符串匹配

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]

提示:

  • 1 <= S.length <= 1000

  • S 只包含字符 "I" 或 "D"。

2、解题思路

  • 双指针法
  • 遇到I则将当前最小的放下
  • 遇到D则将当前最大的放下
class Solution:
    def diStringMatch(self, S: str) -> List[int]:
        res = []
        i = 0
        j = len(S)

        for alpha in S:
            if alpha == "I":
                res.append(i)
                i += 1
            elif alpha == "D":
                res.append(j)
                j -= 1
        res.append(j)
        return res