C语言统计一个字符串中单词的个数

#include <stdio.h>
#include <string.h>

#define SIZE 20

int main()
{
	char str[SIZE]={'\0'};
	int count=0;
	printf("please input the string\n");
	gets(str);
	puts(str);
	int length = strlen(str);
	for (int i=0;i<length;i++)
	{
		//判断是不是空格不是的话在while里面i++执行判断到下一个空格的出现或是结束
		if(str[i]!=' ')
		{
			count++;
			while(str[i]!=' '&&str[i]!='\0')
			{
				i++;
			}
		}
	}
	printf("%d\n",count);
	return 0;

}

#include <stdio.h>
#include <string.h>

#define SIZE 20

int main()
{
	char str[SIZE]={'\0'};
	int count=0;
	int flag=0;
	printf("please input the string\n");
	gets(str);
	puts(str);
	int length = strlen(str);
	for (int i=0;i<length;i++)
	{
		//判断是不是空格，是的话flag=0,
		//不是的话判断前面是不是空格即flag是否等于0，
		//是空格的话说明是新单词的开始
		if(str[i]==' ')
		{
			flag=0;
		}
		else
		{
			if(flag==0)
			{
				count++;
				flag=1;
			}
		}
	}
	printf("%d\n",count);
	return 0;

}