PythonLeetCode(434.字符串中的单词数)

PythonLeetCode(434.字符串中的单词数)

统计字符串中的单词个数，这里的单词指的是连续的不是空格的字符。

请注意，你可以假定字符串里不包括任何不可打印的字符。

示例:

输入: “Hello, my name is John”
输出: 5

Solution:(设置count记录个数，flag记录是否是单词的状态。1.遍历字符串，元素不为空格时，将flag置为1。2.当遇到空格时，count加1，且将flag置为0。3.最后返回的是count加flag的结果。)

class Solution(object):
    def countSegments(self, s):
        """
        :type s: str
        :rtype: int
        """
        count, flag = 0, 0
        
        for i in range(len(s)):
            if s[i] != ' ':
                flag = 1
            elif s[i] == ' ' and flag == 1:
                count += 1
                flag = 0
        return count+flag

solution = Solution()
print(solution.countSegments('Hello, my name is Job'))

5