Question

Formatted question description: https://leetcode.ca/all/332.html

 332	Reconstruct Itinerary

 Given a list of airline tickets represented by pairs of departure and arrival airports [from, to],
 reconstruct the itinerary in order.

 All of the tickets belong to a man who departs from JFK. Thus, the itinerary must begin with JFK.

 Note:

     If there are multiple valid itineraries,
         you should return the itinerary that has the smallest lexical order when read as a single string.
         For example, the itinerary ["JFK", "LGA"] has a smaller lexical order than ["JFK", "LGB"].

     All airports are represented by three capital letters (IATA code).

     You may assume all tickets form at least one valid itinerary.


 Example 1:

 Input: [["MUC", "LHR"], ["JFK", "MUC"], ["SFO", "SJC"], ["LHR", "SFO"]]
 Output: ["JFK", "MUC", "LHR", "SFO", "SJC"]


 Example 2:

 Input: [["JFK","SFO"],["JFK","ATL"],["SFO","ATL"],["ATL","JFK"],["ATL","SFO"]]
 Output: ["JFK","ATL","JFK","SFO","ATL","SFO"]

 Explanation: Another possible reconstruction is ["JFK","SFO","ATL","JFK","ATL","SFO"].
                But it is larger in lexical order.

 @tag-graph

Algorithm

First of all, the graph is established, through the adjacency linked list.

Since the problem requires the solution to be small in alphabetical order, then we consider using multiset, which can be automatically sorted.

After our graph is built, we start traversal from node JFK. As long as there is a node in the multiset mapped by the current node, we take this node out, delete it in the multiset, and then continue to recursively traverse this node.

Finally, the order of our results is the opposite of what we need, we will flip it again at the end。

Code

Java

import java.util.*;

public class Reconstruct_Itinerary {

    public static void main(String[] args) {
        Reconstruct_Itinerary out = new Reconstruct_Itinerary();
        Solution s = out.new Solution();

        System.out.println(s.findItinerary(Arrays.asList(
            Arrays.asList("JFK","SFO"),
            Arrays.asList("JFK","ATL"),
            Arrays.asList("SFO","ATL"),
            Arrays.asList("ATL","JFK"),
            Arrays.asList("ATL","SFO")
        )));
    }


    public class Solution {
        public List<String> findItinerary(List<List<String>> tickets) {

            List<String> result = new ArrayList<>();

            if (tickets == null || tickets.size() == 0) {
                return result;
            }

            // step 1: build the ajdList
            // start => list of destinations
            Map<String, List<String>> adjList = new HashMap<>();
            for (List<String> ticket : tickets) {
                String from = ticket.get(0);
                String to = ticket.get(1);

                if (adjList.containsKey(from)) {
                    adjList.get(from).add(to);
                } else {
                    List<String> neighbors = new ArrayList<>();
                    neighbors.add(to);
                    adjList.put(from, neighbors);
                }
            }

            // step 2: sort the adjlist according to lex order
            for (String from : adjList.keySet()) {
                List<String> neighbors = adjList.get(from);
                Collections.sort(neighbors);
            }

            // step 3: start the dfs
            findItineraryDfs("JFK", adjList, result);

            return result;
        }

        private void findItineraryDfs(String curr, Map<String, List<String>> adjList, List<String> result) {

            List<String> neighbors = adjList.get(curr);

            if (neighbors != null) {
                while (neighbors.size() > 0) {
                    String neighbor = neighbors.get(0); // get first one, lex order
                    neighbors.remove(0); // @note: here is the duplicate removal part, next recursion will not have this neighbour

                    findItineraryDfs(neighbor, adjList, result);
                }
            }

            result.add(0, curr);
        }
    }
}

// Java Lambda
public List<String> findItinerary(String[][] tickets) {
    Map<String, PriorityQueue<String>> targets = new HashMap<>();
    for (String[] ticket : tickets)
        targets.computeIfAbsent(ticket[0], k -> new PriorityQueue()).add(ticket[1]);
    visit("JFK");
    return route;
}

Map<String, PriorityQueue<String>> targets = new HashMap<>();
List<String> route = new LinkedList();

void visit(String airport) {
    while(targets.containsKey(airport) && !targets.get(airport).isEmpty())
        visit(targets.get(airport).poll());
    route.add(0, airport);
}

All Problems

All Solutions