[백준 9019번] DSLR(Python)

문제 링크 : https://www.acmicpc.net/problem/9019

난이도 : 골드 4

출처 : ICPC

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

 

---

📌 풀이 접근 방법

풀이 방법은 어렵지 않았따..

왜냠 아이디어는 엊그제 풀었던 소트게임 문제와 유사했다 !! (dfs 로 풀려고 했다가 봉변당햇던ㅋ)

 

소트게임도 그렇고 이 문제도 bfs로 풀면 된다. 

그 이유는 계속 만들어지는 값을 기준으로 너비 우선 탐색을 하면 가장 먼저 target 값으로 sort되는 경우를 찾을 수 있기 때문!

 

문제 풀이 과정을 보면 

 

1) 입력

• before, after (int 형으로 입력 받기)

2) 중간 과정물

• 방문 표시 visited 필요 -> 시간초과 방지를 위해 set 사용
• 큐에서 command, 숫자를 꺼내면서 확인
• d,s,l,r 에 대한 계산 수행 
• 계산 수행 후 큐에 (이전 command(문자열) +  현재 명령, 숫자) 추가
• 큐 반복

3) 출력

• after 값과 같아지는 순간 command 출력

 

 문제에 설명해주는 거 그대로~ 조건에 따라 코드로 작성해따 (ㅎㅎ 단순 그자체;;)

 

그래서 L, R의 명령어를 처리해줄 때,, 

int -> list 로 바꿔주는 함수를 하나 만들어서

 

1) int 형을 str형으로 바꿔주기

2) 리스트로 변환

3) 리스트로 순서 바꾸기 

 

다음과 같이 아주 복잡하게 처리를 했다 ^^ !

이런 식으로 ㅎ.. 

before2 = register(str(before))
d1,d2,d3,d4 = before2[0], before2[1], before2[2], before2[3]
before2[0], before2[1], before2[2], before2[3] = d4, d1, d2, d3
before2 = int(''.join(map(str, before2)))

 

📌 전체 풀이 코드(시간 초과 발생 😵‍💫)

from collections import deque
import sys

input = sys.stdin.readline

def register(before) :
# 4자리 수가 아닌 경우 0에 대한 처리
    if len(before) == 1: 
        lst = list("000" + before) 
    elif len(before) == 2:
        lst = list("00" + before)
    elif len(before) == 3:
        lst = list("0" + before)
    else :
        lst = list(before)
    return lst


def bfs(before, after) :
    q = deque([('', before)])

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

        if before == after :
            print(command)
            break

        for i in ['D', 'S', 'L', 'R'] :
            if i == 'D' :
                before2 = before * 2 % 10000

            elif i == 'S' :
                before2 = (before - 1) % 10000 

            elif i == 'L' :
                before2 = register(str(before))
                d1,d2,d3,d4 = before2[0], before2[1], before2[2], before2[3]
                before2[0], before2[1], before2[2], before2[3] = d2, d3, d4, d1
                before2 = int(''.join(map(str, before2)))
        
            else:
                before2 = register(str(before))
                d1,d2,d3,d4 = before2[0], before2[1], before2[2], before2[3]
                before2[0], before2[1], before2[2], before2[3] = d4, d1, d2, d3
                before2 = int(''.join(map(str, before2)))

            if before2 not in visited :
                q.append((command + i, before2))
                visited.add(before2)

T = int(input())
for t in range (T) :
    before, after = map(int,(input().split()))
    
    visited = set()
    bfs(before, after)

 

주어진 테케에 대한 정답은 맞지만, 시간 초과가 발생했다.

테케 개수가 엄청나다고 한다.. ^^ 

 

글고 내 코드의 가장 큰 문제점 : 리스트 처리에서 너무 오랜시간이 걸린다.

최적화 시키다가 도저히 모르겠어서 구글링 🔍  

 

L 과 R을 처리해줄 때 

L의 경우 :

천의 자리를 맨 뒤로 보내줘야하기 때문에 1000으로 나눠줬을 때의 몫의 값만 빼놓고, 

나머지 자리수는 10을 곱해줘서 .. 더해준다. 

즉, before2 = before // 1000 + (before % 1000)*10 

 

R의 경우 :

마지막 자리수를 뽑아서 1000을 곱해주고 나머지 3자리를 더해주면 된다. 

즉, before2 = before // 10 + (before % 10) * 1000

 

이거 너무 수학이잖아.. 

 

📌 시간 초과 없는 정답 풀이

from collections import deque
import sys

input = sys.stdin.readline3


def bfs(before, after) :
    q = deque([('',before)])

    while q :
        command, before = q.popleft()
        
        if before == after :
            print(command)
            break

        for i in ['D', 'S', 'L', 'R'] :
            if i == 'D' :
                before2 = before * 2 % 10000
    
            if i == 'S' :
                before2 = (before - 1) % 10000
            
            if i =='L' :
                before2 = before // 1000 + (before % 1000)*10
            
            if i == 'R' :
                before2 = before // 10 + (before % 10) * 1000

            if before2 not in visited :
                q.append((command + i, before2))
                visited.add(before2)

T = int(input())
for t in range (T) :
    before, after = map(int,(input().split()))
    visited = set()
    bfs(before, after)

 

히든케이스가 있는 경우가 있으니 항상 테케가 다 맞다고 정답이 아님에 명심하자..! 

어떻게 하면 시간을 최대한 줄일 수 있을까?! 시간복잡도에 대해서 항상 고민하면서 문제를 풀어야한당

무식하게 코드 작성 금지 ~ 뇌에 힘주고 풀자~

 

 

✅ 문제 풀이 인증

리스트보다 set 쓸 때가 좀 더 빠른거 확인 !!