Java

[백준] 1152 - 단어의 개수

빡성 2024. 7. 18. 23:08

문제
영어 대소문자와 공백으로 이루어진 문자열이 주어진다. 이 문자열에는 몇 개의 단어가 있을까? 이를 구하는 프로그램을 작성하시오. 단, 한 단어가 여러 번 등장하면 등장한 횟수만큼 모두 세어야 한다.
입력
첫 줄에 영어 대소문자와 공백으로 이루어진 문자열이 주어진다. 이 문자열의 길이는 1,000,000을 넘지 않는다. 단어는 공백 한 개로 구분되며, 공백이 연속해서 나오는 경우는 없다. 또한 문자열은 공백으로 시작하거나 끝날 수 있다.
출력
첫째 줄에 단어의 개수를 출력한다.

첫번째 방법은 자바에서 제공하는 trim 메서드를 활용해서 푸는 방법이다.

trim메서드는 문자열의 앞과 뒤에 있는 모든 공백(스페이스, 탭, 줄바꿈 등)을 제거한 새로운 문자열을 반환해준다. 

import java.util.Scanner;

public class Main {
	public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        String sentence = sc.nextLine();
        
        String token[]= sentence.split(" ");
        int answer = token.length;
        
        if (sentence.trim().isEmpty()) {
            System.out.println(answer);
        } else {
            token = sentence.trim().split(" ");
            System.out.println(token.length);
        }
    }
}


두번째 방법은 StringTokenizer을 사용하여 푸는 방법이다. StringTokenizer은 빈 문자열을 토큰으로 인식하지 않기 때문에 문제에서 발생하는 첫번째 문자가 공백이거나 마지막 문자가 공백인 경우를 고려하지 않아도 된다. 따라서 첫번째 방법보다 조금 더 간단하다는 것을 알 수 있다.
하지만 주목해야할 점은 리턴 타입이 다르다는 것이다. 첫번째 방법에서는 리턴 타입이 String[]이고 두번째 방법에서는 리턴 타입이 String이라서 따로 배열을 선언해주지 않아도 된다.

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

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        String sentence = sc.nextLine();

        StringTokenizer st = new StringTokenizer(sentence, " ");
        int token = st.countTokens();

        System.out.println(token);
        
    }
}

'Java' 카테고리의 다른 글

[백준] 11399 - ATM  (0) 2025.01.24
[백준] 10815 - 숫자카드  (0) 2024.07.22
[백준] 1010 - 다리 놓기  (0) 2024.07.15
[자바] 재귀함수 - 하노이 탑 이해하기  (0) 2024.07.11
[자바] 재귀함수 - 개념과 예시  (0) 2024.07.08