[Swift Algorithm] 콜라츠 추측 Programmers(Lv.1)

문제 설명

1937년 Collatz란 사람에 의해 제기된 이 추측은, 주어진 수가 1이 될때까지 다음 작업을 반복하면, 모든 수를 1로 만들 수 있다는 추측입니다. 작업은 다음과 같습니다.

1-1. 입력된 수가 짝수라면 2로 나눕니다.
1-2. 입력된 수가 홀수라면 3을 곱하고 1을 더합니다.
2. 결과로 나온 수에 같은 작업을 1이 될 때까지 반복합니다.

예를 들어, 입력된 수가 6이라면 6→3→10→5→16→8→4→2→1 이 되어 총 8번 만에 1이 됩니다. 위 작업을 몇 번이나 반복해야하는지 반환하는 함수, solution을 완성해 주세요. 단, 작업을 500번을 반복해도 1이 되지 않는다면 –1을 반환해 주세요.

 

제한 조건

• 입력된 수, num은 1 이상 8000000 미만인 정수입니다.

 

입출력 예

n	result
6	8
16	4
626331	-1

 

내 문제 풀이

func solution1(_ num:Int) -> Int {
    var repeatTime = 0 // 콜라츠 추측 반복 횟수 저장 변수
    var n = num
    
    // n이 1이 아닐 때, 짝수라면 2로 나누고 홀수라면 3을 곱하고 1을 더하는 while문
    while n != 1 {
        // 짝수일 때
        if n % 2 == 0 {
            n = n / 2
            repeatTime += 1
        }
        
        // 홀수일 때
        else {
            n = n * 3 + 1
            repeatTime += 1
        }
        
        // 누적된 반복 횟수가 500번이 됐을 경우, 반복 횟수를 -1로 초기화 후 탈출
        if repeatTime == 500 {
            repeatTime = -1
            break
        }
    }
    
    // 콜라츠 추측 성공 시 반복 횟수 출력, 500번 이상 반복 시 -1 리턴
    return repeatTime
}

• 짝수인지 아닌지 판별 하여 콜라츠 추측 계산 적용, 반복 횟수가 500번 됐을 경우 while문을 탈출했다.

 

💡 피드백

• 거의 짝 / 홀 판별 문제나 다름 없었다고 생각하고, 쉬워서 금방 풀었다.

 

---

 

🖍 [ 추가 ] 1주일 후 다시 풀어보기

func solution(_ num:Int) -> Int {
    var n = num
    var count = 0
    
    while n > 1 {
        n%2==0 ? (n=n/2) : (n=n*3+1)
        count += 1
        if count == 500 { count = -1; break }
    }
    
    return count
}

• 첫 번째 풀이와 로직은 같은데, 삼항 연산자 사용하여 훨씬 간결해졌다.

 

---

 

문제

https://programmers.co.kr/learn/courses/30/lessons/12943