(백준) 1152 단어의 개수 C++

문제

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

 

 

입력

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

 

 

출력

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

 

 

문제 풀이

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

int main()
{
    string str;
    getline(cin, str); // 문자열 공백 없이 입력

    int cnt = 0; // 단어 카운트

    for (int i = 0; i < str.length(); i++) {
        if (str[i] == ' ') {
            if (i == str.length()-1) {   // 마지막 공백 무시
                break;
            }
            else {                       // 1,2번 경우) 단어 앞에 공백이 존재하는 경우
                cnt++;
            }
        }
        else {
            if (i == 0) {                // 1번 경우) 맨 처음 단어에 공백이 없는 경우
                cnt++;
            }
        }
    }

    cout << cnt;

    return 0;
}

 

단어를 셀 때 단어 앞의 공백을 기준으로 카운트했고, 두 가지 경우로 나눠 카운트를 진행했다.

1. 제일 앞 단어에 공백이 있거나 없는 경우
2. 중간에 공백이 나오는 경우

앞의 공백을 기준을 카운트하기 때문에 마지막 공백의 유무는 무시해도 된다.

 

 

단어의 개수

1152번: 단어의 개수