2741번: 숫자의 개수

문제 출처 : https://www.acmicpc.net/problem/2577

2577번: 숫자의 개수

언어 : Kotlin

 

문제 설명 :

세 개의 자연수 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까지의 숫자가 각각 몇 번 쓰였는지 차례로 한 줄에 하나씩 출력한다.

 

제한 사항 :

• 시간 제한 : 1초
• 메모리 제한 : 128MB

 

입출력 예 :

입력	출력
150 266 427	3 1 0 2 0 0 0 2 0 0

풀이 : 

import java.io.*

fun main() {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val bw = BufferedWriter(OutputStreamWriter(System.out))

    var arr = intArrayOf()
    while (true) {
        val temp = br.readLine().toInt()
        arr += temp

        if (arr.size == 3) break
    }

    val numCnt = arr.reduce { x, y -> x * y}.toString().groupingBy { it }.eachCount()
    for (i in 0 .. 9) {
        if (numCnt.containsKey(i.digitToChar())) {
            bw.appendLine("${numCnt[i.digitToChar()]}")
        } else {
            bw.appendLine("0")
        }
    }
    bw.flush()
    bw.close()
}