LeetCode 58. 最后一个单词的长度

给定一个仅包含大小写字母和空格 ' ' 的字符串，返回其最后一个单词的长度。
如果不存在最后一个单词，请返回 0 。
说明：一个单词是指由字母组成，但不包含任何空格的字符串。
示例:
输入: "Hello World"
输出: 5

C

int lengthOfLastWord(char * s){
    int count=0,flag=0;
    int len=strlen(s);
    for(int i=len-1;i>=0;i--){
        if(s[i]!=' '){
            flag=1;
            count++;
        }
        else if(s[i]==' '&&flag==1){
            break;
        }
    }
    return count;
}

C++

class Solution {
public:
    int lengthOfLastWord(string s) {
        int cnt = 0;
        int flag = 0;
        for(int i = s.length() - 1; i >= 0; i--) {
            if(flag == 0 && s[i] == ' ') {
                continue;
            }
            if(s[i] != ' ') {
                flag = 1;
                cnt++;
            } else {
                break;
            }
        }
        return cnt;
    }
};