[每日一题] 99. 最后一个单词的长度(字符串、trim函数)

1. 题目来源

链接：最后一个单词的长度
来源：LeetCode

2. 题目说明

给定一个仅包含大小写字母和空格 ’ ’ 的字符串 s，返回其最后一个单词的长度。

如果字符串从左向右滚动显示，那么最后一个单词就是最后出现的单词。

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

说明： 一个单词是指仅由字母组成、不包含任何空格的 最大子字符串。

示例1：

输入: “Hello World”
输出: 5

3. 题目解析

从字符串末尾开始向前遍历，其中主要有两种情况

• 第一种情况，以字符串"Hello World"为例，从后向前遍历直到遍历到头或者遇到空格为止，即为最后一个单词"World"的长度5
• 第二种情况，以字符串"Hello World “为例，需要先将末尾的空格过滤掉，再进行第一种情况的操作，即认为最后一个单词为"World”，长度为5

所以完整过程为先从后过滤掉空格找到单词尾部，再从尾部向前遍历，找到单词头部，最后两者相减，即为单词的长度，时间复杂度：O(n)，n为结尾空格和结尾单词总体长度。

在本题中主要采用trim函数，去掉字符串首尾的空格(连续空格)，在C++中该函数需要自行实现。

4. 代码展示

class Solution {
public:
    void trim(string &s) 
    {
        if (s.empty()) 
        {
            return ;
        }
        s.erase(0,s.find_first_not_of(" "));
        s.erase(s.find_last_not_of(" ") + 1);
    }
    int lengthOfLastWord(string s) {
        reverse(s.begin(), s.end());
        int cnt = 0;
        int flag = 0;
        trim(s);
        for (int len = 0; len < s.size(); ++len) {
            if (s[len] != ' ') {
                cnt++;
            }
            else {
                break;
            }
        }
        return cnt;
    }
};

也有写法如下：

class Solution {
public:
    int lengthOfLastWord(string s) {
        int pos0 = s.find_last_not_of(" ");
        int pos1 = s.find_last_of(" ", pos0);
        if ((pos0 - pos1) < 0)
        {
            return 0;
        }
        return pos0 - pos1;
    }
};