[APS][C++] BOJ B2 1152번 단어의 개수 (feat. getline)

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

---

문제

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

입력

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

출력

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

---

풀이

문자열을 입력받고, 문자열에 포함된 단어의 개수를 세어 출력하면 된다.

• 공백이 포함된 문자열을 받기 위해서 cin 대신 getline 함수 사용
• 공백 한 개만 존재하는 문자열에 대한 예외 처리하기 << 주의!!
• 공백으로 시작하는 문자열이나 공백으로 끝나는 문자열에 대한 에외 처리하기

---

코드

#include <iostream>
#include <string>

using namespace std;

int main() {
  string str;
  getline(cin, str);

  int count = 0;

  if (str.length() == 1 && str[0] == ' ') {
    cout << 0;
    return 0;
  }

  for (int i = 0; i < str.length(); i++) {
    if (i == 0 && str[i] == ' ') continue;
    if (i == str.length() - 1 && str[i] == ' ') continue;
    if (str[i] == ' ') {
      count++;
    }
  }

  cout << count + 1;

  return 0;
}

---

결과