华为OJ：字符串最后一个单词的长度

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

题目描述
计算字符串最后一个单词的长度，单词以空格隔开。
输入描述:
一行字符串，非空，长度小于5000。
输出描述:
整数N，最后一个单词的长度。
输入例子:
hello world
输出例子:
5

//方法一:
#include <iostream>
#include <string.h>
using namespace std;
bool judge(char *str)
{
   bool flag = true;
   char *p = str;
   while(*p!='\0')
   {
      if(*p==' ')
          flag = false;
       p++;
   }
   return flag;
}
int main()
{
    char str[5000];
    char *pre,*pos;
    pre = str;
    while(gets(str))
    {
       if(judge(str))
         {
            cout<<strlen(str)<<endl;
            break;
         }
       while(*pre!='\0')
       {
          pos = pre+1;
          if(judge(pos)) 
               break;
          pre++;
          pos++;
       }
       cout<<strlen(pos)<<endl;
       break;
    } 
    return 0;
}
//方法二
#include<iostream>
#include<string>
#include<vector>

using namespace std;

int main(){
    string input;
    vector<string>arr;
    while(cin>>input){
        arr.push_back(input);
    }
    cout<<arr[arr.size()-1].length()<<endl;    
    return 0;
}