[JAVA/백준] 1152번 - 단어의 개수 풀이

문제

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

입력

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

출력

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

과정

간단한 풀이임에도 몇가지 확인과정에 오류가 발생했던 내용이라 과정을 작성함

첫풀이

import java.util.Scanner;

public class Main {

	public static void main(String[] args) {
		
		Scanner scan = new Scanner(System.in);
		String str = scan.nextLine();
		
		String[] arrStr = (str.trim()).split("\\s");
		
		if()
		System.out.println(arrStr.length);
	}
}

앞뒤 공백부분도 trim 처리를 해주었고 예시문도 예시정답안과 같이 보이는 것을 확인했지만 계속해서 채점이 끝나갈 무렵 틀렸다는 문구가 떴다.

확인해보니 띄어쓰기가 연속으로 두번 작성되면 앞 공백은 trim으로 제거되지만 그 다음 내용없는 부분은 확인이 되지않는 문제. 따라서 isEmpty()을 사용하여 내용이 비어있는 경우 0으로 출력되도록 했다.

풀이1

import java.util.Scanner;

public class Main {

	public static void main(String[] args) {
		
		Scanner scan = new Scanner(System.in);
		String str = scan.nextLine();
		
		str = str.trim();
		
		if(str.isEmpty()==false) {
			String[] arrStr = str.split("\\s");
			System.out.println(arrStr.length);
		} else {
			System.out.println(0);
		}
	}
}

더불어 풀이과정에 split과 trim이 아닌 StringTokenizer을 사용하는 방법도 있어 함께 기재한다. 속도 면에서 풀이2가 훨씬 앞서는것으로 보인다

풀이2(권장)

import java.util.Scanner;
import java.util.StringTokenizer;

public class Main{

	public static void main(String[] args) {
				
		Scanner scan = new Scanner(System.in);
		String str = scan.nextLine();
		
		StringTokenizer st = new StringTokenizer(str, " ");
		System.out.println(st.countTokens());
	}
}