Length of Last Word

Leetcode 面试题

简单题。但不知道本地运行结果正确的程序，跑到oj上输出结果却不对。但以下代码是正确的。模拟题一个，照着要求做就好。

class Solution {
public:
    int lengthOfLastWord(const char *s) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        if(s[0]=='\0') return 0;
        int ans = 0, cnt = 0;
        for(int i = 0; s[i]!='\0'; i ++){
            if(s[i]==' '){
                if(cnt != 0) ans = cnt;
                cnt = 0;
            }else{
                cnt ++;
            }
        }
        if(cnt != 0) ans = cnt;
        return ans;
    }
};