33-alstjr7437

[alstjr7437]

🔗 문제 링크

DSLR

자~ 웃으세요~~!

✔️ 소요된 시간

15분 + 시간초과 문제로 10분 ㄷㄷ..

✨ 수도 코드

평소의 BFS하면 좌표를 건드리는 문제 였는데 이번 문제는 비슷하지만 DSLR 각 커맨드에 따라 숫자를 변경하면 되겠더라구요!

딱 숫자도 10000까지니까 그냥 Visited배열로 만들어서 사용했습니다.

1. A와 결과 문자열을 먼저 queue에 넣는다
2. queue Pop을 하면서 반복문과 함수를 이용해서 DSLR의 숫자를 각각 넣는다
3. visited == False면 true로 바꾸고 숫자와 현재까지 cmd를 넣는다.
4. 만약 cur == B로 결과값이 나왔으면 현재까지 command를 출력한다!

from collections import deque
import sys

input = sys.stdin.readline

T = int(input())

def command(register, cmd):
    if cmd == 0:
        return ((register * 2) % 10000, "D")
    elif cmd == 1:
        return (9999 if register == 0 else register - 1, "S")
    elif cmd == 2:
        return (register % 1000 * 10 + register // 1000, "L")
    elif cmd == 3:
        return (register // 10 + (register % 10) * 1000, "R")

for _ in range(T):
    A, B = map(int, input().split())

    visited = [False] * 10000
    visited[A] = True

    queue = deque([(A, "")])
    while queue:
        cur, result = queue.popleft()
        
        if cur == B:
            print(result)
            break
        
        for i in range(4):
            register, cmd = command(cur, i)
            if not visited[register]:
                visited[register] = True
                queue.append((register, result + cmd))

📚 새롭게 알게된 내용

이게 사람은 적응의 동물이라고 계속해서 함수 쓰는데
Swfit를 요즘 너~~무 많이써서
func로 해버리고 : 를 넣어야하는데 {}를 넣고 컨텍스트 스위칭이 잘안되네요...

그리고 Python오래만이라 시간초과 생각을 안해서 왜 시간초과 나지 계속 코드 만져보다가...
Pypy로 바꾸니까 됐습니다!!

[tgyuuAn]

from collections import deque
import sys

def input(): return sys.stdin.readline().rstrip()
for _ in range(int(input())):
    A, B = map(int, input().split())

    deq = deque()
    visited = {A,}
    deq.append((A, ""))
    while deq:
        now_value, history = deq.popleft()

        if now_value == B:
            print(history)
            break

        for query in range(4):
            if query == 0:
                new_value = int(now_value*2%10000)
            
                if new_value in visited: continue
        
                new_history = history+"D"
                
            elif query == 1:
                new_value = now_value-1 if now_value != 0 else 9999

                if new_value in visited: continue
                
                new_history = history+"S"
    
            elif query == 2:
                str_now_value = str(now_value)
                while len(str_now_value) < 4:
                    str_now_value = "0" + str_now_value
                
                new_value = int(str_now_value[1]+str_now_value[2]+str_now_value[3]+str_now_value[0])
    
                if new_value in visited: continue
                
                new_history = history+"L"
    
            else:
                str_now_value = str(now_value)
                while len(str_now_value) < 4:
                    str_now_value = "0" + str_now_value
                new_value = int(str_now_value[3]+str_now_value[0]+str_now_value[1]+str_now_value[2])

                if new_value in visited: continue

                new_history = history+"R"
        
            visited.add(new_value)
            deq.append((new_value, new_history))

대놓고 BFS라고 문제에 써져있는 느낌이라 따로 코멘트는 안달겠습니다!!

근데 그나저나 코드 가독성이 굉장히 좋네요

[H0ngJu]

저도 시간초과 때문에 코드 수정하다가 pypy3로 하니까 통과되네용
민석님 코드를 보니까 차라리 처음부터 cmd 4개를 묶은 함수를 만드는게 더 깔끔했을 것 같아요!! 👍👍

수고하셨슴다 ~~

import sys
from collections import deque

def input() : return sys.stdin.readline().rstrip()

T = int(input())
test = [list(map(int, input().split())) for _ in range(T)]

def mod_D(n):
    return (2 * n) % 10000

def mod_S(n):
    return 9999 if n == 0 else n - 1

def mod_L(n):
    d1 = n // 1000
    tmp = n % 1000
    return (tmp * 10) + d1

def mod_R(n):
    d4 = n % 10
    tmp = n // 10
    return (d4 * 1000) + tmp

for A, B in test:
    q = deque([(A, "")])
    visited = [False] * 10000
    visited[A] = True

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

        if num == B:
            print(cmd)
            break

        d_num = mod_D(num)
        if not visited[d_num]:
            visited[d_num] = True
            q.append((d_num, cmd + "D"))

        s_num = mod_S(num)
        if not visited[s_num]:
            visited[s_num] = True
            q.append((s_num, cmd + "S"))

        l_num = mod_L(num)
        if not visited[l_num]:
            visited[l_num] = True
            q.append((l_num, cmd + "L"))

        r_num = mod_R(num)
        if not visited[r_num]:
            visited[r_num] = True
            q.append((r_num, cmd + "R"))