문제 출처: programmers.co.kr/learn/courses/30/lessons/12943
코딩테스트 연습 - 콜라츠 추측
1937년 Collatz란 사람에 의해 제기된 이 추측은, 주어진 수가 1이 될때까지 다음 작업을 반복하면, 모든 수를 1로 만들 수 있다는 추측입니다. 작업은 다음과 같습니다. 1-1. 입력된 수가 짝수라면 2
programmers.co.kr
1. 문제 이해
입력된 수가 짝수라면 2로 나눔
홀수라면 *3 + 1
주어진 수가 1이될때까지 반복
다음작업이 반복되는 횟수 리턴
500번 반복해도 안되면 -1출력
num은 8백만 이하
2. 풀이 계획
반복문 or 재귀
재귀면 전역변수 쓰거나 cnt인자 넘겨줘야함
반복문이 재실행되거나 재귀 호출때마다 cnt += 1
cnt > 500이면 -1출력
그전에 num이 1이되면 반복문나와서 cnt 출력
3. 구현
# 반복문으로 푼 풀이
def solution(num):
cnt = 0
while True:
if cnt > 500 :
return -1
if num == 1:
return cnt
if num % 2 :
num = num * 3 + 1
else :
num //= 2
cnt += 1
# 재귀로 푼 풀이
cnt = 0
def solution(num) :
global cnt
cnt += 1
if cnt > 500 :
return -1
if num == 1 :
return cnt - 1 # 주의 : 맨처음에 1을 더해주므로 마지막에 1을 뺀값을 return해야함
if num % 2:
return solution(num * 3 + 1)
else :
return solution(num // 2)
4. 검토 및 개선
'Problem Solving > 프로그래머스' 카테고리의 다른 글
Lv1. 행렬의 덧셈 [O] (0) | 2020.10.30 |
---|---|
Lv1. 하샤드 수 [O] (0) | 2020.10.28 |
Lv1. 최대공약수와 최소공배수 [O!] (0) | 2020.10.28 |
Lv1. 키패드 누르기 [X] (0) | 2020.10.28 |
Lv1. 짝수와 홀수 [O] (0) | 2020.10.27 |
댓글