Programmers / 2단계 / 숫자 변환하기 / python

 

코딩테스트 연습 - 숫자 변환하기 | 프로그래머스 스쿨 (programmers.co.kr)

 

 

나의 풀이 (실패)

문제는 쉬워보였지만 막상 적절한 접근 방법이 떠오르지 않았다.

def solution(x, y, n):
    num=[]
    count=0
    x1=x; x2=x; x3=x
    
    while x3<y:
        x1+=n
        x2*=2
        x3*=3
        num.append(x1)
        num.append(x2)
        num.append(x3)
        count+=1
        
        if y in num:
            return count
            
    return -1

 

모범 답안

def solution(x, y, n):
    answer = 0
    dp = set()
    dp.add(x)
    
    while dp:
        if y in dp:
            return answer
        else:
            dp_y = set()
            for i in dp:
                if i+n <= y:
                    dp_y.add(i+n)
                if i*2 <= y:
                    dp_y.add(i*2)
                if i*3 <= y:
                    dp_y.add(i*3)
            dp = dp_y
            answer += 1
            
    return -1

 

dp를 활용한 문제였다.
x를 y로 변환하기 위한 최소 연산의 수가 필요하기 때문에 set 을 활용했다.

1. x 값을 dp에 우선 대입한다.
2. y 가 dp에 있다면, 정답을 리턴
3. dp를 돌면서, i+n i*2 i*3 3가지 연산 중 y보다 작거나 같으면 dp_y라는 set에 add한다.
4. 3번 연산을 마친 dp_y에 dp를 copy하면서 (answer+1)하면서 1~3 과정을 반복한다.

 

참조

프로그래머스 - 숫자 변환하기 (velog.io)

프로그래머스 - 숫자 변환하기