[백준] 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 →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로 변환하기 위해 필요한 최소한의 명령어 나열을 출력한다. 가능한 명령어 나열이 여러가지면, 아무거나 출력한다.

 

📚 문제 풀이

해당 문제는 BFS를 통해 수의 변화과정을 탐색해 나갔다.

큐에 변화된 숫자 뿐만 아니라 그 숫자가 되기 위한 명령어도 같이 저장해주어 갱신하였다.

D를 수행하기 위해서는 단순히 (2*n)%10000 을 수행하면 됐다. S는 n이 0이 아닐때는 n-1 을 해주었고 0일때는 n값에 9999를 넣었다.

L과 R 같은 경우에는 n이 4자리 숫자일 경우 문자열 인덱싱을 해주어 명령어를 수행했고, 4자리 숫자가 아닐 경우 0을 포함시켜 인덱싱을 해주었다.

 

💻 코드

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

# BFS 정의
def bfs(a,b):
    queue=deque()
    queue.append([a,'']) # 현재 숫자와 명령어
    visited=[False]*10000 # 방문배열
    while queue:
        tob,pro=queue.popleft()
        if tob==b: # 현재 숫자가 b와 같다면 명령어를 리턴
            return pro
        # S 수행    
        if tob == 0:
            S=9999
        else:
            S=tob-1
		
        # L과 R 수행
        if len(str(tob)) == 4:
            L = int(str(tob)[1:]+str(tob)[0])
            R = int(str(tob)[-1]+str(tob)[:-1])
        else:
            L = tob*10
            R = int(str(tob)[-1]+str(0)*(4-len(str(tob)))+str(tob)[:-1])

		# 이동 가능한 숫자 배열
        move=[(2*tob)%10000, S, L, R]
        # 명령어
        way=['D','S','L','R']
        for i in range(4):
            if not visited[move[i]] and move[i] < 10000: # 방문하지 않았고 10000 이하라면 방문
                visited[move[i]] = True # 방문처리
                queue.append([move[i], pro+way[i]]) # 큐에 숫자와 명령어 삽입

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

 

🍭 시행착오

시초와 틀렸습니다를 반복하였던...ㅎㅎ

틀렸던 이유는 L과 R이 4자리 숫자가 아닐때 처리과정에 문제가 있었다.

그래서 백준에도 글을 남겼다. 혹시 나같은 실수를 하고 있는 분이 있을수도 있으니

https://www.acmicpc.net/board/view/96558

글 읽기 - 계속 7%에서 틀리시는 분들을 위해 글을 적습니다ㅎ

첫번째 맞았습니다는 방문처리를 dict 자료구조를 사용해서 하였고, 두번째 맞았습니다에서는 방문처리를 list 자료구조를 사용해서 하였고 또 이동제한을 10000이 아닌 100000로 0 하나를 더 붙혔어서 그부분도 수정해주었었다. 3번째는 0하나 더 붙힌거 수정하고 dict 자료구조를 사용하였다.

 

🔗 문제링크 DSLR 

9019번: DSLR