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