[백준/파이썬][Gold IV] DSLR - 9019

[Gold IV] DSLR - 9019

문제 링크

성능 요약

메모리: 216180 KB, 시간: 5032 ms

분류

너비 우선 탐색, 그래프 이론, 그래프 탐색

제출 일자

2025년 2월 4일 21:18:26

문제 설명

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

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

위에서 언급한 것처럼, 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로 변환하기 위해 필요한 최소한의 명령어 나열을 출력한다. 가능한 명령어 나열이 여러가지면, 아무거나 출력한다.

풀이

from collections import deque

def main() :
    A, B = map(int, input().split())

    visited = [0] * 10001

    q = deque()
    q.append([A, ''])
    visited[A] = 1

    while q :
        num, command = q.popleft()

        if num == B:
            print(command)
            break

        d = num * 2 % 10000
        if not visited[d] :
            visited[d] = 1
            q.append([d, command + 'D'])

        s = (num - 1) % 10000
        if not visited[s] :
            visited[s] = 1
            q.append([s, command + 'S'])

        l = num // 1000 + (num % 1000) * 10
        if not visited[l] :
            visited[l] = 1
            q.append([l, command + 'L'])

        r = num // 10 + (num % 10) * 1000
        if not visited[r] :
            visited[r] = 1
            q.append([r, command + 'R'])

T = int(input())
for _ in range(T) :
    main()

 

그래프 탐색을 이런식으로 활용하는 것도 재미있는 발상인 것 같다. 최대 숫자인 1만개의 visited 배열을 만들고 해당 인덱스에 방문할 때 마다 방문 표시로 0 => 1 로 갱신하였다. 

나중에 실제로 그래프를 직접 그려보면서 풀이해보면 좋을 것 같다. 

생각해보니 어차피 중복될 가능성이 없기 때문에 이런 식으로 적용할 수 도 있구나라는 걸 느끼게 되었다.