LeetCode 58 [Length of Last Word]

原题

给定一个字符串， 包含大小写字母、空格' '，请返回其最后一个单词的长度。如果不存在最后一个单词，请返回 0 。

样例
给定 s = "Hello World"，返回 5 。

解题思路

• 按" "分开长字符串，得到得到一个单词数组
• 反转数组，遍历，返回第一个不是空串的单词的长度

完整代码

class Solution(object):
    def lengthOfLastWord(self, s):
        """
        :type s: str
        :rtype: int
        """
        if not s:
            return 0
            
        words = s.split(" ")[::-1]
        for word in words:
            if len(word) != 0:
                return len(word)
        return 0