[Swift｜코드카타] (프로그래머스) 입문 #23. 콜라츠 추측 - 함수 2개 쓰기

🔴 문제

문제 - 콜라츠 추측 (클릭)

 

 

🔵 풀이 1

주어진 수를 담은 변수 'temp'가 1이 될 때까지 반복되는 while문을 만들면 될 것 같다. -> temp != 1 인 동안 반복하는 while문 작성

 

몇 번 돌았는지를 체크하고 반환하는 'count' 변수도 있어야 하겠다.

문제를 보면 500번째 반복까지는 괜찮아보인다. 'count'가 501인 순간 반복문을 break하고 -1을 return하게 하면 된다.

 

func solution(_ num:Int) -> Int {
    guard num != 1 else { return 0 }
    var temp = num
    var count = 0
    
    while temp != 1 {
        count += 1
        if count > 500 { break }
        
        if temp % 2 == 0 {  // collatz 연산 수행
            temp /= 2
        } else {
            temp = temp * 3 + 1
        }
        
        if temp == 1 { return count } // 연산 결과가 1이면 count 반환
    }
    return -1 // count > 500 일 경우 반환
}

 

 

🔵 풀이 2 - collatz 연산 새로운 함수로 분리

다른 풀이를 보니, 콜라츠의 연산 부분을 다른 함수로 묶은 경우가 몇몇 보였다.

나도 짝수이면 2로 나누고, 홀수이면 3을 곱하고 1을 더하는 부분을 'collatz'라는 함수로 따로 만들어보겠다.

// collatz 함수 정의
func collatz(_ num:Int) -> Int {
    if num % 2 == 0 {
        return num / 2 // 짝수면 2로 나눈 값 반환
    } else {
        return num * 3 + 1 // 홀수면 3을 곱하고 1을 더한 값 반환
    }
}

func solution(_ num:Int) -> Int {
    guard num != 1 else { return 0 } // num이 1이면 0 반환
    var temp = num
    var count = 0
    
    while temp != 1 {
        count += 1
        if count > 500 { break }
        temp = collatz(temp) // collatz 연산 수행
        if temp == 1 { return count } // 연산 결과가 1이면 count 반환
    }
    return -1
}

 

지금은 collatz함수를 만들기 전과 후의 차이가 별로 안 보이지만, 

연산이 더 복잡하다면, 함수를 따로 생성하는 편이 코드 가독성을 높이는 데에 도움이 될 것 같다.