알고리즘 연습/DFS와 BFS

[🥈1 / 백준 12852 / 파이썬] 1로 만들기 2

김세진 2021. 9. 24. 02:01
반응형

 

 

12852번: 1로 만들기 2

첫째 줄에 1보다 크거나 같고, 106보다 작거나 같은 자연수 N이 주어진다.

www.acmicpc.net

 

문제

정수 X에 사용할 수 있는 연산은 다음과 같이 세 가지 이다.

  1. X가 3으로 나누어 떨어지면, 3으로 나눈다.
  2. X가 2로 나누어 떨어지면, 2로 나눈다.
  3. 1을 뺀다.

정수 N이 주어졌을 때, 위와 같은 연산 세 개를 적절히 사용해서 1을 만들려고 한다. 연산을 사용하는 횟수의 최솟값을 출력하시오.

입력

첫째 줄에 1보다 크거나 같고, 106보다 작거나 같은 자연수 N이 주어진다.

출력

첫째 줄에 연산을 하는 횟수의 최솟값을 출력한다.

둘째 줄에는 N을 1로 만드는 방법에 포함되어 있는 수를 공백으로 구분해서 순서대로 출력한다. 정답이 여러 가지인 경우에는 아무거나 출력한다.

 

예제 입력 1 

2

예제 출력 1 

1
2 1

예제 입력 2 

10

예제 출력 2 

3
10 9 3 1

 

풀이

 

주어진 숫자를 1로 만드는 최소 연산 횟수를 찾되, 그 방법까지 출력하는 문제이다.

dp로도 해결할 수 있지만, BFS를 이용했다.

BFS를 활용하여 큐에 삽입할 때, 거쳐온 경로까지 삽입하는 방법을 사용했다.

 

from collections import deque

n = int(input())
q = deque()
q.append((n,[n]))
visited = [0]*(n+1)

while(q):
    num,ans = q.popleft()
    if num == 1:
        print(len(ans)-1)
        print(*ans)
        break
    if not visited[num]:
        visited[num]=1
        if num%3==0:
            q.append((num//3,ans+[num//3]))
        if num%2==0:
            q.append((num//2,ans+[num//2]))
        q.append((num-1,ans+[num-1]))
반응형