HJ1 字符串最后一个单词的长度

题目描述
计算字符串最后一个单词的长度，单词以空格隔开。

输入描述:
一行字符串，非空，长度小于5000。
输出描述:
整数N，最后一个单词的长度。

#include <iostream>
using namespace std;
int main()
{
    int ans = 0;
    string str;
    getline(cin, str);
    for(int i = 0; i < str.size(); i++)
        if(str[i] == ' ')
            ans = 0;
        else
            ans++;
    cout << ans;
    return 0;
}