华为机试 — 字符串最后一个单词的长度

题目描述

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

输入描述:

一行字符串，非空，长度小于5000。

输出描述:

整数N，最后一个单词的长度。

示例1

输入

hello world

输出

5

#include<iostream>
#include<string>
using namespace std;
int main() {
    string str1, str2;
    getline(cin, str1);
    int len_str1 = str1.length();
    str2 = str1;
    for (int i = 0; i < len_str1; i++) {
        if (str1[i] == ' ' && i < len_str1 - 1){
            str2 = str1.substr(i+1,len_str1-i-1);
        }
    }
    cout<<str2.length();
    return 0;
}