[백준/BOJ] 1152번 단어의 개수 (C/C++)

백준 온라인 저지(BOJ) 1152번 단어의 개수

https://www.acmicpc.net/problem/1152

1152번: 단어의 개수

* 사용언어 : C언어, C++

 

1. 문제

영어 대소문자와 공백으로 이루어진 문자열이 주어짐

1,000,000 을 넘지 않는 문자열에 있는 단어의 개수를 출력

(단어는 공백 1개로만 구분되고, 공백은 연속해서 나오지 않음)

 

2. 풀이

scanf 함수로 문자열을 "%s" 로 입력 받으면 공백에서 끊기게 됩니다.

 

이 현상을 해결하며 풀려고 scanf 함수의 return 값 등을 찾아보다가,

점점 억지스러운 코드가 되는 것 같아서 구글링을 한 뒤 좋은 방법을 찾을 수 있었습니다.

"%[^\n]" : \n 이 나올 때 까지 전부 입력받기

\n 은 enter 로 항상 입력의 가장 마지막에 있으므로 모든 공백을 함께 입력받을 수 있습니다.

이 방법으로 전체를 입력 받았습니다.

 

 

그 후 단어의 개수를 세는 방법을 고민했는데,

모든 Case 를 한 번에 반영하려하니 방향이 잘 잡히지 않았습니다.

 

그래서 일단 나머지 Case 는 전부 무시하고

앞 뒤에 공백이 없는 1번 입력 예제만 생각하면서 구현을 시작했습니다.

예제 입력 1

대전제) 공백 이후에는 반드시 1개의 단어가 있다

규칙 1) cnt 는 1로 시작 (공백이 없는 첫 단어를 Count 하지 못하므로)
규칙 2) 문자열 탐색 중 공백(' ') 이 있으면 ++cnt (공백 뒤 단어 1개)

 

 

이 방식으로 1번 예제는 쉽게 통과하지만, 2번 예제는 통과하지 못합니다.

왜냐하면 첫 단어가 2번 Count 되기 때문입니다. (cnt 를 1로 시작했는데, 공백으로 추가로 +1)

예제 입력 2

따라서 첫 번째 글자가 공백이 아닐 때만 cnt 를 1로 시작하게 했습니다.

(탐색하면서 규칙 2) 에 의해 Count)

 

 

끝에 공백이 있는 3번 예제는 공백 이후 단어가 없으니,

최초 가정에 위배되므로 탐색을 마친 후 아래와 같이 처리했습니다.

예제 입력 3

규칙 3) 탐색 종료 후 마지막 글자가 공백(' ') 이면 --cnt

 

이로써 3가지 예제를 모두 알맞게 출력할 수 있게 되었습니다.

 

추가로 입력이 공백으로 딱 1글자로 주어질 수 있는데,

규칙 3) 에 의해 공백 후 단어가 없는 것으로 자연스럽게 처리되므로 보완하지 않았습니다.

 

3. 코드

#include <stdio.h>

int main() {
	char s[1000001];
	scanf("%[^\n]", &s);

	int cnt = 0;
	if (s[0] != ' ') ++cnt;

	int i = 0;
	while (s[i]) {
		if (s[i] == ' ') ++cnt;
		++i;
	}

	if (s[i - 1] == ' ') --cnt;

	printf("%d\n", cnt);
	return 0;
}