[C언어] 1152번 - 단어의 개수

문제

영어 대소문자와 공백으로 이루어진 문자열이 주어진다. 이 문자열에는 몇 개의 단어가 있을까? 이를 구하는 프로그램을 작성하시오. 단, 한 단어가 여러 번 등장하면 등장한 횟수만큼 모두 세어야 한다.

입력

첫 줄에 영어 대소문자와 공백으로 이루어진 문자열이 주어진다. 이 문자열의 길이는 1,000,000을 넘지 않는다. 단어는 공백 한 개로 구분되며, 공백이 연속해서 나오는 경우는 없다. 또한 문자열은 공백으로 시작하거나 끝날 수 있다.

출력

첫째 줄에 단어의 개수를 출력한다.

코드

#include <stdio.h>
#include <string.h>
#define len 1000000

int main(void) {
	
	char str[len] = {};
	int cnt = 0;
	int i = 0;
	
	scanf("%[^\n]", str); //줄바꿈까지 문자열 입력 

	if(str[0] == ' ') { //첫 글자가 공백이라면 두번째 글자부터 본다
		i = 1;
	}

	while(str[i] != '\0') { //배열에 각 인덱스의 문자가 null일 때까지 반복
		if(str[i] == ' ') { //공백을 찾았다면 개수를 세준다
			cnt++;
		}
		
		i++;
	}

	if(str[i - 1] == ' ') { //마지막 글자가 공백이라면 
		printf("%d", cnt);
	}
	else { //마지막 글자에 공백이 없다면
		printf("%d", cnt + 1);
	}
	
	return 0;
}