Leetcode-58: Length of Last Word

最新推荐文章于 2024-09-20 00:30:00 发布

原创最新推荐文章于 2024-09-20 00:30:00 发布 · 169 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Leetcode

algorithm-design 专栏收录该内容

817 篇文章

订阅专栏

本文介绍了一种使用C++编程语言来寻找并计算给定字符串中最后一个非空单词长度的方法。通过将输入字符串反转，并查找首个非空字符的位置来实现。特别考虑了边缘情况，如空字符串和尾部含有多个空格的字符串。

这题较易，但陷阱也不少。

我的解法是把string s倒置过来，找第一个非空字符串。

注意:

1) "a "的情况。此时要用一个while()遍历所有的空格。
2) " "的情况，此时要注意pos不能超界。

#include <iostream>
#include <string>
#include <algorithm>

using namespace std;

int lengthOfLastWord(string s) {
    if (!s.size()) return 0;
    string s_orig(s);

    reverse(s.begin(), s.end());
    size_t pos = s.find(' ');

    if (pos==string::npos) return (int)(s_orig.size());
    else if (pos==0) { //for the case of "a " or "a     "
        while((pos<s.size()-1) && (s.at(++pos)==' '));   //pos<s.size() is for the case of " "
        size_t pos2 = s.find(' ', pos);

        if (pos2==string::npos) return s.size()-pos;
        else return pos2-pos;
    }
    else
        return (int)pos;

}

int main()
{
    string input_s="Hello World";
    cout<<lengthOfLastWord(input_s)<<endl;

    input_s="a    ";
    cout<<lengthOfLastWord(input_s)<<endl;

    input_s=" ";
    cout<<lengthOfLastWord(input_s)<<endl;

    return 0;
}