[백준/9019/파이썬] DLSR

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

9019번: DSLR

 

Problem

네 개의 명령어 D, S, L, R 을 이용하는 간단한 계산기가 있다. 주어진 서로 다른 두 정수 A와 B(A ≠ B)에 대하여 A를 B로 바꾸는 최소한의 명령어를 생성하는 프로그램을 작성하세요.

 

조건

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이 된다.
5. 출력은 명령어로 출력된다. 예를들어, A=1234,B=3412 -> LL 이다.
6. 프로그램 입력은 T 개의 테스트 케이스로 구성된다. 테스트 케이스 개수 T 는 입력의 첫 줄에 주어진다. 각 테스트 케이스로는 두 개의 정수 A와 B(A ≠ B)가 공백으로 분리되어 차례로 주어지는데 A는 레지스터의 초기 값을 나타내고 B는 최종 값을 나타낸다. A 와 B는 모두 0 이상 10,000 미만이다.

SOL)

 

1. D,L,S,R 연산을 함수형태로 구현하였다. 이때,calL,calD연산은 처음에 log를 이용해서 연산해줬는데, n=0일때, log0은 정의가 되지 못하기 때문에, 연산으로 대체하였음 
2. 그리고, 연산 ->명령어를 바꾸는 부분을, 문자열을 dq에 넣고, dq에 있는 문자열에 추가하는 형식으로 구현하였습니다.

주의)

문제에서 n의 네 자릿수를 d1, d2, d3, d4라고 하자(즉 ,n = ((d1 × 10 + d2) × 10 + d3) × 10 + d4라고 하자) 라고 하였기 때문에, 4자리를 기준으로 calL,calD를 만들어야 합니다. 저는 N자리를 기준으로 만들어서 계속 틀렸었습니다.(사실, n자리를 기준으로 만드는게 맞는거 같은데) 예를 들어,123->231을 해보면, L이 아니라, SSDSRSL 이라는 값이 나오게됩니다.

 

from collections import deque

tc = int(input())
def calD(n):
    res=n*2
    if res>9999:
        res%=10000
    return res

def calS(n):
    res=n
    if res==0:
        return 9999
    res -=1
    return res

def calL(n):
    front = n%1000
    back = n//1000
    res = front*10 + back
    #print("front:",front,"back:",back,"res:",res)
    return res

def calR(n):
    front = n%10
    back = n//10
    res=front*1000+back
    #print("front:",front,"back:",back,"res:",res)
    return res

def solve(s,t):
    dq=deque()
    visited = set()
    dq.append((s,""))
    visited.add(s)
    while dq:
        x,oper=dq.popleft()
        if x == t:
            print(oper)
            return
        tmp = calD(x)
        if tmp not in visited:
            visited.add(tmp)
            dq.append((tmp,oper+"D"))
        tmp = calS(x)
        if tmp not in visited:
            visited.add(tmp)
            dq.append((tmp,oper+"S"))
        tmp = calL(x)
        if tmp not in visited:
            visited.add(tmp)
            dq.append((tmp,oper+"L"))
        tmp = calR(x)
        if tmp not in visited:
            visited.add(tmp)
            dq.append((tmp,oper+"R"))
            
while tc:
    A,B = map(int,input().split())
    solve(A,B)
    tc-=1