자바(Java) [백준] 1152번 : 단어의 개수

문제

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

입력

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

출력

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

---

처음에 제출한 코드가 틀려서 질문 게시판을 찾아보니!!!

공백 입력시 0이 나와야 한다는 답변을 보고 머리를 쳤다.

 답변 주신 분 감사합니다.

해서 코드를 끄적끄적 고치고 난 뒤 정답 확인!

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        String str = sc.nextLine();
        String str2 = str.trim();
        int cnt = 0;
        for(int i = 0; i < str2.length(); i++) {
            if(str.length() == 1 && str.charAt(i) == ' ') break;
            else if(str2.charAt(i) == ' ') cnt++;
        }
        if(str.length() == 1 && str.charAt(0) == ' ')System.out.println(cnt);
        else System.out.println(cnt + 1);
    }
}