[프로그래머스][LEVEL3] 여행경로

# 문제 원문

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

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

주어진 항공권을 모두 이용하여 여행경로를 짜려고 합니다. 항상 "ICN" 공항에서 출발합니다.

항공권 정보가 담긴 2차원 배열 tickets가 매개변수로 주어질 때, 방문하는 공항 경로를 배열에 담아 return 하도록 solution 함수를 작성해주세요.

제한사항

• 모든 공항은 알파벳 대문자 3글자로 이루어집니다.
• 주어진 공항 수는 3개 이상 10,000개 이하입니다.
• tickets의 각 행 [a, b]는 a 공항에서 b 공항으로 가는 항공권이 있다는 의미입니다.
• 주어진 항공권은 모두 사용해야 합니다.
• 만일 가능한 경로가 2개 이상일 경우 알파벳 순서가 앞서는 경로를 return 합니다.
• 모든 도시를 방문할 수 없는 경우는 주어지지 않습니다.

입출력 예

tickets / return

[["ICN", "JFK"], ["HND", "IAD"], ["JFK", "HND"]]	["ICN", "JFK", "HND", "IAD"]
[["ICN", "SFO"], ["ICN", "ATL"], ["SFO", "ATL"], ["ATL", "ICN"], ["ATL","SFO"]]	["ICN", "ATL", "ICN", "SFO", "ATL", "SFO"]

 

# 문제 풀이

 

입력 받은 tickets를 모두 순회하면서 그 경로를 짜는 문제이다. 문제 조건으로 모든 티켓을 사용해야한다는 점에 주의해서 방문 경로들을 탐색하고 그렇게 구한 방문 경로를 알파벳 기준으로 정렬해서 반환하는 문제이다. 방문 경로를 구하는 단계에서 알파벳 순으로 정렬해서 그리디로 접근 할 경우 테스트케이스 1,2번의 먼저 알파벳 순으로 정렬한 경우 경로 끝에 이어지지 않는 함정 케이스에 걸려서 시간을 허비하게 된다.

개인적으로 FP으로 짤려고 반복문을 활용한 그리디 방식으로 짰더니 1번 2번 테스트케이스를 통과하지 못해서 시간을 허비했다. 실제 테스트시에는 주의하자. 지문에 그리디를 암시하는 내용이 없다면 안하는게 맞을거 같다.

 

# 솔루션 플로우

1. 입력 받은 tickets를 dfs 탐색한다.

2. 시작(current)는 ICN에서 시작하고 입력 받은 tickets에서 current에 해당하는 모든 경우를 탐색해서 구한다.

3. 구해진 routes이 복수라면 알파벳순의 정렬한다.

4.  구해진 result를 반환한다.

1. DFS

function solution(tickets) {
    const result = [];

    const dfs = (arr, current, stack = ['ICN']) => {
        if (stack.length === tickets.length + 1) {
            result.push([...stack]);
        }
        if (arr.length > 0) {
            arr.forEach((route, idx) => {
                if (route[0] === current) {
                    stack.push(route[1]);
                    dfs(
                        [
                            ...arr.slice(0, idx),
                            ...arr.slice(idx + 1, arr.length)
                        ],
                        route[1],
                        stack
                    );
                    stack.pop();
                }
            });
        }
    };
    dfs(tickets, 'ICN');

    return result.sort((a, b) => a.join('') > b.join('') ?
        1 :
        -1
    )[0];
}