Leetcode 58: Length of Last Word

leetcode_58b.cpp

#include <iostream>
#include <string>

using namespace std;

int lengthOfLastWord(string s) {
    int n = s.size();
    int i = n-1;
    while (i >= 0 && s[i]==' ') i--; //skip trailing whitespace
    int last = i;
    for ( ; i >= 0; --i) {
        if (s[i] == ' ') return n-(i+1);
    }
    return last-i;
}

int main() {
    string s = "Hello World";
    cout << lengthOfLastWord(s) << "\n";
    return 0;
}