原题地址:https://leetcode-cn.com/problems/reverse-words-in-a-string/description/
题目描述:
给定一个字符串,逐个翻转字符串中的每个单词。
示例:
输入: "the sky is blue",
输出: "blue is sky the".
说明:
无空格字符构成一个单词。
输入字符串可以在前面或者后面包含多余的空格,但是反转后的字符不能包括。
如果两个单词间有多余的空格,将反转后单词间的空格减少到只含一个。
进阶: 请选用C语言的用户尝试使用 O(1) 空间复杂度的原地解法。
解题方案:
这题看似简单,却花了我大量的时间找bug。。主要是这几个问题:
- 函数str.erase(it),每次删除之后,it会自动加一,移到原先删除节点的下一个节点。
- string类型内部反转,reverse(s.begin(), s.end())。
- 字符串结尾处需要额外处理,比较容易出错。
代码:
class Solution {
public:
void reverseWords(string &s) {
if(s.length() == 0)
return ;
reverse(s.begin(), s.end());
int flag = 0;
auto it = s.begin(), pre = s.begin();
while(*it == ' ' && it != s.end())
s.erase(it);
while(it != s.end()){
if(*it == ' ' && flag == 1){
s.erase(it);
continue;
}
else if(*it == ' ' && flag == 0){
flag = 1;
reverse(pre, it);
}
else if(flag == 1){
pre = it;
flag = 0;
}
it ++;
}
if(flag == 0)
reverse(pre, it);
else
s.erase(s.end() - 1);
}
};