[Python] 백준 5014 & 프로그래머스 오픈채팅방

#5012 스타트링크

# 총 F층
# 스타트링크 G층
# 현재 강호 S층
# U는 위로 u층
# D는 아래로 d층

F, S, G, U, D = map(int, input().split())
visited = [-1] * (F + 1)

from collections import deque
def bfs():
    q = deque()
    q.append(S)
    visited[S] = 0
    while q:
        x = q.popleft()
        if x == G:
            return visited[x]
        for nx in (x+U, x-D):
            if 1 <= nx <= F and visited[nx] == -1:
                q.append(nx)
                visited[nx] = visited[x] + 1

    return "use the stairs"

print(bfs())

 

record = ["Enter uid1234 Muzi", "Enter uid4567 Prodo","Leave uid1234","Enter uid1234 Prodo","Change uid4567 Ryan"]

def solution(record):
    memo = {}
    answer = []
    for message in record:
        info = message.split()
        if info[0] == "Enter" or info[0] == "Change":
            memo[info[1]] = info[2]

    for id in record:
        info = id.split()
        if info[0] == "Enter":
            answer.append(memo[info[1]]+"님이 들어왔습니다.")
        elif info[0] == "Leave":
            answer.append(memo[info[1]]+"님이 나갔습니다.")

    return answer
print(solution(record))
# memo[info[0]] = info[2]