算法学习——leetcode58. 最后一个单词的长度

题目：

给定一个仅包含大小写字母和空格 

' '

如果不存在最后一个单词，请返回 0 。

说明：一个单词是指由字母组成，但不包含任何空格的字符串。

示例:

输入: "Hello World"
输出: 5

解题思路：

用空格来标识一个单词的结束，并将每个单词的length存入到向量lengths中

代码：

[code]class Solution {
public:
int lengthOfLastWord(string s) {
int length = 0;
vector<int> lengths;
lengths.push_back(length);
for(int i=0;i<s.length();i++){
if(s.at(i) == ' '){
if(length>0)
lengths.push_back(length);
length=0;
}
else if(i == s.length()-1 && s.at(i) != ' '){
length++;
lengths.push_back(length);
}
else
length++;
}
// lengths.push_back(length);
return lengths.back();
}
};

百度之后发现可以先去掉首尾的空格，代码可以简化一些，但是内存使用和用时好像也没有很大的变化

代码:

[code]class Solution {
public:
int lengthOfLastWord(string s) {
s.erase(s.find_last_not_of(" ") + 1);
s.erase(0, s.find_first_not_of(" "));
int length = 0;
for(int i=0;i<s.length();i++){
if(s.at(i) != ' '){
length++;
}
else
length = 0;
}
return length;
}
};