백준 1152

백준 1152 단어의 개수

🔍 문제 설명

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

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

✔ 입력

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

✔ 출력

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

💡 풀이

공백에 대한 예외처리를 잘 해주는 것이 핵심이다.
첫 공백, 마지막 공백 에 대해 생각해보고, 단어의 갯수와 공백의 갯수의 관계에 대해 생각해보면 해결할 수 있다.

📝 소스코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {
    static FastReader scan = new FastReader();
    static String a;

    public static void main(String[] args) {
        input();
        pro();
    }

    private static void pro() {
        if (a.length() == 1 && a.charAt(0) == ' ') {
            System.out.println(0);
            return;
        }

        int count = 0;
        for (int i = 0; i < a.length(); i++) {
            if (i == 0 && a.charAt(0) == ' ') continue; // 첫번째 공백
            if (i == a.length() - 1 && a.charAt(i) == ' ') continue; // 맨 마지막 줄
            if (a.charAt(i) == ' ') count++;
        }

        System.out.println(count + 1);
    }

    private static void input() {
        a = scan.nextLine();
    }

    static class FastReader {
        BufferedReader br;
        StringTokenizer st;

        public FastReader() {
            br = new BufferedReader(new InputStreamReader(System.in));
        }

        String nextLine() {
            String str = "";
            try {
                str = br.readLine();
            } catch (IOException e) {
                e.printStackTrace();
            }
            return str;
        }
    }
}