[Swift Algorithm] 숫자의 개수 BOJ #2577

문제 설명

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

 

입출력 예제

입력

150
266
427

출력

3
1
0
2
0
0
0
2
0
0

 

내 문제 풀이

import Foundation

let A = Int(readLine()!)!
let B = Int(readLine()!)!
let C = Int(readLine()!)!
let multi = String(A*B*C)
var result = Array(repeating: 0, count: 10)

for i in multi {
    result[Int(String(i))!] += 1
}

print(result.map({String($0)}).joined(separator : "\n"))

• 세 가지 입력을 모두 곱한 값을 String으로 변환해주었다.
• 0부터 9까지 숫자의 개수를 카운트하기 위해 10칸짜리 result 배열을 생성했다.
• multi 문자열을 순서대로 검사하며 result 배열의 해당하는 위치에 1씩 더해주었다.
• 개행으로 출력을 구분하기 위해 joined()함수에 "\n" 구분자를 사용했다.

 

💡 피드백

• joined 함수 separator를 사용하는 방법만 알고 있다면 쉬운 문제였다.

 

---

 

문제

https://www.acmicpc.net/problem/2577