给定只含 "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"
。
思路:我们设置small表示左边最小的数,初始化为0,big表示最大的数,初始化为S.length(),那么只要保证遇到I时把最小的数加进去,遇到D时把最大的数加进去,那么最后拼凑出来的数组一定是对的。
参考代码:
class Solution {
public:
vector<int> diStringMatch(string S) {
int small = 0, big = S.length();
vector<int> res;
for (int i = 0; i < S.length(); i++) {
if (S[i] == 'I') res.push_back(small++);
else res.push_back(big--);
}
res.push_back(small);
return res;
}
};