[Python] 백준 9019번 - DSLR

https://www.acmicpc.net/problem/9019

9019번: DSLR

혼자 힘으로 풀었는가? X

알고리즘 분류
 - 그래프 이론
 - 그래프 탐색
 - 너비 우선 탐색(BFS)

 

문제

네 개의 명령어 D, S, L, R 을 이용하는 간단한 계산기가 있다. 이 계산기에는 레지스터가 하나 있는데, 이 레지스터에는 0 이상 10,000 미만의 십진수를 저장할 수 있다. 각 명령어는 이 레지스터에 저장된 n을 다음과 같이 변환한다. n의 네 자릿수를 d1, d2, d3, d4라고 하자(즉 n = ((d1 × 10 + d2) × 10 + d3) × 10 + d4라고 하자)

1. D: D 는 n을 두 배로 바꾼다. 결과 값이 9999 보다 큰 경우에는 10000 으로 나눈 나머지를 취한다. 그 결과 값(2n mod 10000)을 레지스터에 저장한다.
2. S: S 는 n에서 1 을 뺀 결과 n-1을 레지스터에 저장한다. n이 0 이라면 9999 가 대신 레지스터에 저장된다.
3. L: L 은 n의 각 자릿수를 왼편으로 회전시켜 그 결과를 레지스터에 저장한다. 이 연산이 끝나면 레지스터에 저장된 네 자릿수는 왼편부터 d2, d3, d4, d1이 된다.
4. R: R 은 n의 각 자릿수를 오른편으로 회전시켜 그 결과를 레지스터에 저장한다. 이 연산이 끝나면 레지스터에 저장된 네 자릿수는 왼편부터 d4, d1, d2, d3이 된다.

위에서 언급한 것처럼, L 과 R 명령어는 십진 자릿수를 가정하고 연산을 수행한다. 예를 들어서 n = 1234 라면 여기에 L 을 적용하면 2341 이 되고 R 을 적용하면 4123 이 된다.

여러분이 작성할 프로그램은 주어진 서로 다른 두 정수 A와 B(A ≠ B)에 대하여 A를 B로 바꾸는 최소한의 명령어를 생성하는 프로그램이다. 예를 들어서 A = 1234, B = 3412 라면 다음과 같이 두 개의 명령어를 적용하면 A를 B로 변환할 수 있다.

1234 →L 2341 →L 3412
1234 →R 4123 →R 3412

따라서 여러분의 프로그램은 이 경우에 LL 이나 RR 을 출력해야 한다.

n의 자릿수로 0 이 포함된 경우에 주의해야 한다. 예를 들어서 1000 에 L 을 적용하면 0001 이 되므로 결과는 1 이 된다. 그러나 R 을 적용하면 0100 이 되므로 결과는 100 이 된다.

입력

프로그램 입력은 T 개의 테스트 케이스로 구성된다. 테스트 케이스 개수 T 는 입력의 첫 줄에 주어진다. 각 테스트 케이스로는 두 개의 정수 A와 B(A ≠ B)가 공백으로 분리되어 차례로 주어지는데 A는 레지스터의 초기 값을 나타내고 B는 최종 값을 나타낸다. A 와 B는 모두 0 이상 10,000 미만이다.

출력

A에서 B로 변환하기 위해 필요한 최소한의 명령어 나열을 출력한다. 가능한 명령어 나열이 여러가지면, 아무거나 출력한다.

 

 

---

 

처음엔 어떻게 풀지... 싶다가 DSLR의 모든 과정을 큐에 담으면 되겠다! 라는 생각으로 BFS로 자연스럽게 풀게 되었다.

 

우선 이문제에서 중요한 몇가지가 있다.

 

123을 L 하면 1230

123을 R 하면 3012 이다. 312가 아닌

 

또한 큐에서 pop한 값 value가 0일 때 9999로 바꾸는 것이지

value - 1 == 0 일 때 9999 (즉, value == 1일 때) 바꾸는 게 아니다.

 

이 두 가지를 조심하면 되는데 난 결국 문제를 제대로 이해하지 못해서 못 풀고 다른 사람의 풀이법을 참조하게 된 것이다.

나는 두 번째의 value가 0일 때 바꿨어야 했는데 value가 1일 때 바꾸는 방식으로 해서 계속 문제가 틀렸습니다. 를 보게 되었다.

 

import sys
input = sys.stdin.readline
from collections import deque

t = int(input())
  
for _ in range(t):
  a, b = map(int, input().split())
  q = deque()
  visited = [False] * 10001
  visited[a] = True
  q.append((a, ''))
  while q:
    value, letter = q.popleft()
    if value == b:
      print(letter)
      break

    D = 2*value%10000
    if visited[D] == False:
      q.append((D, letter+'D'))
      visited[D] = True

    if value == 0:
      S = 9999
    else:
      S = value-1
    if visited[S] == False:
      q.append((S, letter+'S'))
      visited[S] = True

    L = 10*(value%1000) + value//1000
    if visited[L] == False:
      q.append((L, letter+'L'))
      visited[L] = True

    R = 1000*(value%10) + value//10
    if visited[R] == False:
      q.append((R, letter+'R'))
      visited[R] = True

코드 제출은 PyPy3로 제출했다. Python3는 시간초과가 발생하였다.

 

다음은 내가 참고한 반례들이다.

 

12
5 0 RD
0 7 SDRDRDRDDSDR
9 0 SDSRSD
47 63 RSDLS
26 60 RSSRR
60 50 RSL
32 38 RSLSSSD
98 366 SRDLSD
5310 6 DRSSR
7519 2006 SDDRS
7579 4005 SRDDSDR
2720 8031 DDRSRDDS

git 백준 9019번 - DSLR.py