Java/백준

[1152] 단어의 개수

tmd1 2022. 6. 30. 22:11

문제:

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

 

1152번: 단어의 개수

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

www.acmicpc.net

코드:

import java.util.*;

public class Main {

	public static void main(String[] args) {
		// TODO Auto-generated method stub

		Scanner sc = new Scanner(System.in);
		String testCase = sc.nextLine();
		
		String [] strArr = testCase.split("\\s"); // 공백 기준으로 나눔
		
		if(testCase.equals(" ")) { // 공백만 입력 됐을 시
			System.out.println("0");
		}else {
			if(strArr[0].equals("")) {
				System.out.println(strArr.length-1); // 맨 처음 부분이 공백일 경우 공백도 배열에 들어감
			}else {                                  // 따라서 -1
				System.out.println(strArr.length);
			}
		}
		
		
		
	}

}

 

공백만 입력 되었을 경우 strArr[0]이 존재하지 않아 오류가 발생할 수 있으므로 따로 나누어 처리