Length of Last Word

比较简单的问题：

#include<iostream>
#include<string>
using namespace std;

int main()
{
    int lengthOfLastWord(const char *s);
    char a[20];
    cin.getline(a,20);
    cout << lengthOfLastWord(a) << endl;
}

int lengthOfLastWord(const char *s) {
        int i=0,new_word=0,last_length=0;
        while(s[i]!='\0'){
            if(s[i]==' ')
                new_word=1;
            else if(new_word==1 && s[i]!=' '){
                last_length=1;
                new_word=0;
            }
            else{
                ++last_length;
            }
            ++i;
        }
        return last_length;
    };