计算输入的单词数有多少个

#include <stdio.h>
#include <stdlib.h>
int fun(char *s)
{
    int counter=0,flag=1;
    while(*s)   //遍历数组
    {
        if(*s==' ') //一个空格代表一个单词结束
            flag=1;
        else if(flag==1)  //标记为真就代表一个单词开始
        {
            counter++;
            flag=0;     //修改标记进行下一个判断
        }
        s++;    //指针往下移动一个
    }
    return counter;
}
int main()
{
    char str[100];
    gets(str);
   printf("%d",fun(str));
    return 0;
}