본문 바로가기

Algorithm/Baekjoon

(Java) [Baekjoon 2577 - 숫자의 개수] - 2024. 3. 7.(목)

문제  

세 개의 자연수 A, B, C가 주어질 때 A × B × C를 계산한 결과에 0부터 9까지 각각의 숫자가 몇 번씩 쓰였는지를 구하는 프로그램을 작성하시오.

예를 들어 A = 150, B = 266, C = 427 이라면 A × B × C = 150 × 266 × 427 = 17037300 이 되고, 계산한 결과 17037300 에는 0이 3번, 1이 1번, 3이 2번, 7이 2번 쓰였다.

입력

첫째 줄에 A, 둘째 줄에 B, 셋째 줄에 C가 주어진다. A, B, C는 모두 100보다 크거나 같고, 1,000보다 작은 자연수이다.

출력

첫째 줄에는 A × B × C의 결과에 0 이 몇 번 쓰였는지 출력한다. 마찬가지로 둘째 줄부터 열 번째 줄까지 A × B × C의 결과에 1부터 9까지의 숫자가 각각 몇 번 쓰였는지 차례로 한 줄에 하나씩 출력한다.


BOJ2577.java

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.stream.Collectors;

public class BOJ2577 {
    public static void main(String[] args) {
        try (BufferedReader br = new BufferedReader(new InputStreamReader(System.in))) {
            final int first = Integer.parseInt(br.readLine());
            final int second = Integer.parseInt(br.readLine());
            final int third = Integer.parseInt(br.readLine());
            int mul = first * second * third;
            final int[] numbers = new int[10];
            while (mul > 0) {
                numbers[mul%10]++;
                mul /= 10;
            }
            final String result = Arrays.stream(numbers).mapToObj(String::valueOf).collect(Collectors.joining("\n"));
            System.out.println(result);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

 

처음 문제를 보고 바로 떠올렸던 방법이다.

% 연산자를 통하여 각 자릿수를 배열에 기록하는 방법이다.

이렇게 풀고 다른 방법이 없나 생각하다가 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;

public class BOJ2577 {
    public static void main(String[] args) {
        try (BufferedReader br = new BufferedReader(new InputStreamReader(System.in))) {
            final int first = Integer.parseInt(br.readLine());
            final int second = Integer.parseInt(br.readLine());
            final int third = Integer.parseInt(br.readLine());
            final int[] numbers = new int[10];
            String.valueOf(first * second * third).chars().forEach(c -> numbers[c - '0']++);
            Arrays.stream(numbers).forEach(System.out::println);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

forEach와 stream, 메소드 참조를 사용하여 다른 방법도 사용했다.

숫자의 크기가 크지 않아서 그런가 시간과 메모리 차이는 나지 않았다.

 

출처 : Baekjoon online judge, https://www.acmicpc.net/problem/2577