[Algorithm] 자릿수 더하기 (map, reduce)

🖥 문제

자연수 N이 주어지면, N의 각 자릿수의 합을 구해서 return 하는 solution 함수를 만들어 주세요.
예를 들어 N = 123이면 1 + 2 + 3 = 6을 return 하면 됩니다.

🤔 풀이 방법

for문으로 구현

1. String(n)
   • Int 형식으로 되어있는 n을 String으로 바꾸어준다.
2. answer += Int(String(strN))!
   • strN을 한글자씩 Int 형식으로 바꾸어주며 더해준다.

map으로 구현

1. String(n)
   • Int 형식으로 되어있는 n을 String으로 바꾸어준다.
2. .map{Int(String($0))!} : map을 이용하여 n을 [Int] 형식으로 바꾸어준다.
   • $0 : [String.Element]
   • String($0) : String
   • Int(String($0))! : [Int]
3. .reduce(0, +)
   • [Int] 배열 안의 숫자를 각각 꺼내어 + 연산한다.

👩🏻‍💻 Swift Code

for문으로 구현

import Foundation

func solution(_ n:Int) -> Int
{
    var answer:Int = 0

    for strN in String(n) {
        answer += Int(String(strN))!
    }

    return answer
}

map으로 구현

import Foundation

func solution(_ n:Int) -> Int
{
    var answer:Int = String(n).map{Int(String($0))!}.reduce(0, +)

    return answer
}

 

코딩테스트 연습 - 자릿수 더하기