티스토리 뷰

코딩테스트

Lv.0 swift 1로 만들기

알롱도담쓰 2024. 3. 29. 16:30

문제 설명

정수가 있을 때, 짝수라면 반으로 나누고, 홀수라면 1을 뺀 뒤 반으로 나누면, 마지막엔 1이 됩니다. 예를 들어 10이 있다면 다음과 같은 과정으로 1이 됩니다.

10 / 2 = 5
(5 - 1) / 2 = 2
2 / 2 = 1
위와 같이 3번의 나누기 연산으로 1이 되었습니다.

정수들이 담긴 리스트 num_list가 주어질 때, num_list의 모든 원소를 1로 만들기 위해서 필요한 나누기 연산의 횟수를 return하도록 solution 함수를 완성해주세요.

 

제한 조건

3 ≤ num_list의 길이 ≤ 15
1 ≤ num_list의 원소 ≤ 30

내 답안

import Foundation

func solution(_ num_list:[Int]) -> Int {
    var count = 0
    var one = 0
    
    for i in num_list {
        one = i
        
        while one != 1 {
            one = one % 2 == 0 ? one / 2 : (one - 1) / 2
            count += 1
        }
    }

    return count
}

 

접근 방법

  1. 각 요소가 1이 될 때까지 while 문 돌리기

 

다른 풀이

func solution(_ num_list:[Int]) -> Int {
    func dfs(_ num: Int, _ count: Int) -> Int {
        if num == 1 { return count }
        return dfs(num / 2, count + 1)
    }
    return num_list.reduce(0) { $0 + dfs($1, 0) }
}

 

재귀함수를 쓸 수도 있다!! 

 

알게 된 것

 

'코딩테스트' 카테고리의 다른 글

Lv.0 swift 수열과 구간 쿼리 3  (0) 2024.03.28
Lv.0 swift 문자열 겹쳐쓰기  (1) 2024.03.27
Lv.0 swift 분수의 덧셈  (0) 2024.03.27
Lv.0 swift 배열 만들기 5  (0) 2024.03.25
Lv.0 swift 날짜 비교하기  (0) 2024.03.25
최근에 올라온 글