用C语言判断输入单词的个数

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<stdlib.h>
int count(char ch[])
{
    int i;
    int m=0;
    int flag=1;
    for(i=0;ch[i]!='\0';i++)    //i am a student
    {
        if(ch[i]==' ')
        {
            flag=1;
        }
        else
        {
            if(flag==1)
            {
                flag=0;
                m++;
            }
        }
    }
    return m;
}
void main()
{
    char str[100];
    printf("请输入一个字符串长度小于100！：\n");
    gets(str);
    printf("源字符串为：");
    puts(str);
    printf("单词个数是：%d\n",count(str));
}