[Programmers/Go] 콜라츠 추측

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 미만인 정수입니다.

2. 풀이

// 콜라츠 추측
package collatz

func Solution(num int) int {
	cnt := 0

	for num > 1 && cnt < 500 {
		if num%2 == 0 {
			num /= 2
		} else {
			num = num*3 + 1
		}
		cnt++
	}

	if cnt == 500 {
		return -1
	}

	return cnt
}

3. 테스트

package collatz

import "testing"

func TestCollatz(t *testing.T) {
	input := []int{6, 16, 626331}
	expect := []int{8, 4, -1}

	for i := range input {
		result := Solution(input[i])
		if result != expect[i] {
			t.Errorf("Test%d: Wrong result", i+1)
			t.Error("expect:", expect[i])
			t.Error("result:", result)
		}
	}
}

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

코딩테스트 연습 - 콜라츠 추측