[BOJ] 1152번 - 단어의 개수

 

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

1152번: 단어의 개수

문제

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

입력

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

출력

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

---

1. 어떻게 풀까?

1-1) 단어의 개수 = 띄어쓰기의 개수 + 1

문장에서 단어의 개수는 안의 띄어쓰기의 개수에 하나를 더한 값이라 생각해도 된다. (ex. 2개의 띄어쓰기 = 3개의 단어)

공백을 포함하여, 문장 전체를 받은 다음, 띄어쓰기의 개수를 하나씩 세주면 된다는 의미이다.

따라서, 문장의 string에서 하나씩 char형 중 공백(' ')의 개수를 세어주면 된다. 다만...

1-2) 띄어쓰기의 개수 ≠ 공백의 개수

공백의 개수가 띄어쓰기의 개수를 보장하지 못한다. 문제의 입력을 잘 읽어보면 '문자열은 공백으로 시작하거나 끝날 수 있다.' 로 공백으로 시작하거나 끝날 수 있기 때문이다. 따라서, 양 끝이 공백인지 확인해주는 작업이 필요하다.

 

또한, 만약 빈 문자열이 입력으로 들어온다면 이러한 방법은 통하지 않게 된다. 이를 예외로 처리해줌으로써 오답을 피해갈 수 있다. 문장 전체를 입력으로 받아주고, 이러한 논리를 구현하면 손쉽게 맞았습니다! 를 받을 수 있다.

2) 최종 코드

#include <bits/stdc++.h>
using namespace std;
 
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
 
    string str; getline(cin, str);
    int str_length = str.length();
    int cnt = 1;
 
    if (str.empty()) {
        cout << "0";
        return 0;
    }
 
    for (int i = 0; i < str_length; i++) {
        if (str[i] == ' ') {
            cnt++;
        }
    }
    
    if (str[0] == ' '){ cnt--; }
 
    if (str[str_length - 1] == ' '){ cnt--; }
 
    cout << cnt << "\n";
 
    return 0;
}