[Baekjoon/C++] 단어의 개수 1152번

Baekjoon Online Judge

문제로 이동

 

문제

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

 

입력

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

 

출력

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

 

---

예제 입력	예제 출력
The Curious Case of Benjamin Button	6
(공백)The first character is a blank	6
The last character is a blank(공백)	6

---

풀이

#include <iostream>
#include <string>

using namespace std;

int main() {
    // 공백 포함해서 입력받기
    string S;
    getline(cin, S);
    
    // 공백 개수 찾기
    int count = 0;
    for (int i = 0; i < S.length(); i++) {
        if (S[i] == ' ') {
            if(i != 0 && i != S.length() - 1)
                count++;
        }
    }

    // 공백만 있는 경우
    if (count == 0 && S.length() == 1 && S[0] == ' ')
        cout << 0;
    // 단어의 수는 맨 앞과 맨 끝의 공백을 제외한 공백의 수 + 1과 같다.
    else 
        cout << count + 1;

    return 0;
}

공백을 포함해서 입력받기 위해 <string>의 getline을 사용하였다.

단어의 개수는 단어 사이사이에 있는 공백의 수 + 1과 같으므로 for문에서 if(i != 0 && i != S.length() - 1) 이라는 제한을 뒀다.

출력 부분에서 공백만 입력된 경우 단어가 1이라고 표시되는 것을 막기 위해 단어의 길이 등을 확인하는 조건문을 달았다.