[백준 / C++] 1152번 : 단어의 개수 - B2

관련 문제

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

문제 설명

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

입력

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

출력

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

내 코드

#include <iostream>
#include <string>
using namespace std;

int main(void)
{
    int result = 0;
    string S;
    getline(cin, S);

    if (S.empty() || (S.size() == 1 && S[S.size() - 1] == ' '))
    {
        cout << result << endl;
        return 0;
    }

    int startIdx = S[0] == ' ' ? 1 : 0;
    int endIndx = S[S.size() - 1] == ' ' ? S.size() - 1 : S.size();

    for (int i = startIdx; i < endIndx; i++)
        result = S[i] == ' ' ? result + 1 : result;

    cout << ++result << endl;
    return 0;
}

배울 점

cin, getc 등과 같은 입력 방식으로는 공백이 있는 문자열 입력 불가

getline - 공백이 있는 문자열 한 줄 입력
ex) getline(cin, string)