[Algorithm] 백준 1152번 : 단어의 개수

문제

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

 

입력

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

 

출력

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

 

풀이

먼저 주어진 문자열에서 단어가 몇개인지 보는것인데

문자열을 charAt()을 이용해서 ' ' 공백일때를 개수를 세면된다

하지만 이때 생각해야하는 경우의 수가 있느데

문자열 맨앞과 맨뒤가 공백이 아니면 cnt에 +1을 해주고

맨앞과 맨뒤가 공백이면 cnt에 -1을 해주고

맨앞 또는 맨뒤중 하나만 공백이면 cnt에 변화를 안줘도 된다

 

나는 charAt()을 이용해서 풀었지만 더 간단하게는

StringTokenizer 클래스를 이용하면 된다

문자열을 우리가 지정한 구분자로 문자열을 쪼개주는 클래스이다

 

코드

1) charAt()을 이용해서

import java.util.Scanner;

public class Main {
	public static void main(String[] args) {

		// 문장이 주어지면 이 문장에 단어가 몇개인지 알아내기

		Scanner sc = new Scanner(System.in);

		String str = sc.nextLine();

		int cnt = 0;

		//
		for (int i = 0; i < str.length(); i++) {
			if (str.charAt(i) == ' ') {
				cnt++;
			}
		}

		// 첫 번째와 마지막이 공백이 아닐때
		if (str.charAt(0) != ' ' && str.charAt(str.length() - 1) != ' ') {
			cnt++;
		}
		// 첫 번째와 마지막이 공백인 경우
		if (str.charAt(0) == ' ' && str.charAt(str.length() - 1) == ' ') {
			cnt--;
		}
		System.out.println(cnt);
	}
}

 

2) StringTokenizer를 이용해서

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

public class Main {
	public static void main(String[] args) {

		// 문장이 주어지면 이 문장에 단어가 몇개인지 알아내기

		Scanner sc = new Scanner(System.in);
		
		String s = sc.nextLine();

		StringTokenizer st = new StringTokenizer(s, " ");

		System.out.println(st.countTokens());
	}
}