[프로그래머스 level3 여행경로] bfs/dfs 활용하기

해결해야하는 문제는 다음과 같았다.

https://programmers.co.kr/learn/courses/30/lessons/43164

코딩테스트 연습 - 여행경로

 

이러한 아이디어를 바탕으로 작성한 코드는 다음과 같다.

 

from collections import deque

def solution(tickets):
    answer = deque()
    ticket_dict = {}

    for s,d in tickets :
        if s not in ticket_dict.keys():
            ticket_dict[s] = [d]
        else:
            ticket_dict[s].append(d)

    keys = ticket_dict.keys()
    for i in keys:
        ticket_dict[i].sort(reverse=True)

    stack = ['ICN']
    while stack:
        dt = stack[-1]
        if dt not in keys :
            answer.appendleft(stack.pop())
        elif not ticket_dict[dt]:
            answer.appendleft(stack.pop())
        else :
            stack.append(ticket_dict[dt].pop())


    return list(answer)