[백준/BOJ] gold4 - 9019번 DSLR (Python)

▶9019 - DSLR

백준 로고

▶문제

네 개의 명령어 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 → L2341 → L3412
1234 → R4123 → R3412

따라서 여러분의 프로그램은 이 경우에 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로 변환하기 위해 필요한 최소한의 명령어 나열을 출력한다. 가능한 명령어 나열이 여러 가지면, 아무거나 출력한다.

 

▶풀이

일단 python3로 제출해서 시간초과가 났다.

그래서 pypy3로 새로 제출했는데 이건 통과가 됐다.

아직은 시간복잡도 면에서 조금 부족한 코드인 것 같다.

 

bfs와 queue를 이용해 풀어주면 되는 문제이다.

queue에는 현재 값과 DSLR을 알려주는 string 두 가지를 넣어주면 된다.

 

다른 bfs와 거의 똑같이 풀어도 pypy3에서는 정답이 나온다.

from collections import deque


def bfs():
    queue = deque()
    queue.append((a, ''))
    visit = [0 for _ in range(10000)]
    visit[a] = 1

    while queue:
        tmp, string = queue.popleft()
        if tmp == b:
            return string
        
        
        tmp2 = (tmp * 2) % 10000
        if visit[tmp2] == 0:
            visit[tmp2] = 1
            queue.append((tmp2, string + 'D'))

        tmp2 = (tmp - 1) % 10000
        if visit[tmp2] == 0:
            visit[tmp2] = 1
            queue.append((tmp2, string + 'S'))

        tmp2 = tmp // 1000 + (tmp % 1000) * 10
        if visit[tmp2] == 0:
            visit[tmp2] = 1
            queue.append((tmp2, string + 'L'))

        tmp2 = (tmp % 10) * 1000 + tmp // 10
        if visit[tmp2] == 0:
            visit[tmp2] = 1
            queue.append((tmp2, string + 'R'))


t = int(input())
answer = []

for _ in range(t):
    a, b = map(int, input().split())
    answer.append(bfs())

for ans in answer:
    print(ans)