题目来源:链接: [https://leetcode-cn.com/problems/di-string-match/comments/].
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]
输出:"DDI"
输出:[3,2,0,1]
提示:
1. 1 <= S.length <= 1000
2. S 只包含字符 "I" 或 "D"。
2.解决方案
我的代码为什么提交不对啊。。。。逻辑没错啊啊啊
一开始没弄懂题目到底是啥意思,后来根据示例慢慢找出规律。(读题得有耐心)
第一次代码 忽略了 返回值 是 vector<int ,直接返回了首地址 编译错误
第二次代码 测试时每次的结果全部都少一个,纠结了好久发现原来是 n 的问题,n - -之后,循环次数明显变少了。
希望自己下次细心!!!
第一遍的代码如下
class Solution {
public:
vector<int> diStringMatch(string S) {
int n = S.length();
int i = 0; // 建议1:i 建议在for循环体内 定义并使用
int count = 0;
for(i = 0; i < n; i++)
{
if( S[i] == 'I' )
{
S[i] = count;
count++;
}
if( S[i] == 'D')
{
S[i] = n;
n--; // 错误1: n每次-- 会导致循环少一次 与实际结果不合
}
}
S[i] = count;
return S; //错误2: 注意题目要求的返回值
}
};
3.代码
最终提交的代码如下:
class Solution {
public:
vector<int> diStringMatch(string S) {
int n = S.length();
int min = 0;
int max = n;
vector<int> res;
for(int i = 0; i < n; i++)
{
if( S[i] == 'I' )
{
res.push_back(min++);
//min++;
}
else if( S[i] == 'D') //这里可以优化些 直接else
{
res.push_back(max++);
//max--;
}
}
res.push_back(min);
return res;
}
};
执行用时: 64 ms, 在DI String Match的C++提交中击败了10.95% 的用户
内存消耗: 11.1 MB, 在DI String Match的C++提交中击败了0.86% 的用户
2019/2/27 胡云层 于南京 4