942. 增减字符串匹配

https://leetcode-cn.com/problems/di-string-match/

给定只含 “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”。


题解
对题目的理解关键在于

如果 S[i] == “I”,那么 A[i] < A[i+1]
如果 S[i] == “D”,那么 A[i] > A[i+1]

当I出现时,总有n存在,使得等式成立,长度为N的S最多有N个I,故出现I时总选择未用的数字里最小的,才能使后面的I出现时有数可用
同理适用于D,当D出现时总要选择当前未用的数字里最大的


C++

class Solution {
public:
    vector<int> diStringMatch(string S) {
        int N=S.size();
        int l=0,r=N;
        vector<int> res;
        for(char ch:S){//只要ch是char型就继续循环
            if(ch=='I'){
                res.push_back(l++);//先置入l,再l=l+1,++l正好相反
            }
            else{
                res.push_back(r--);
            }
        }
        res.push_back(l);//res实际大小为N+1,故再将最后一个未使用数字置入,显然此时有r==l
        return res;
    }
};

PYTHON

class Solution(object):
    def diStringMatch(self, S):
        """
        :type S: str
        :rtype: List[int]
        """
        l=0
        r=len(S)
        res=[]
        for i in S:
            if i=='I':
                res.append(l)
                l=l+1
            else:
                res.append(r)
                r=r-1
        res.append(l)
        return res
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值