diff --git a/src/main/java/com/fishercoder/solutions/_332.java b/src/main/java/com/fishercoder/solutions/_332.java index 6713d6e365..e963c0b6da 100644 --- a/src/main/java/com/fishercoder/solutions/_332.java +++ b/src/main/java/com/fishercoder/solutions/_332.java @@ -12,12 +12,12 @@ public static class Solution1 { /** * credit: https://discuss.leetcode.com/topic/36383/share-my-solution */ - public List findItinerary(String[][] tickets) { + public List findItinerary(List> tickets) { Map> flights = new HashMap<>(); LinkedList path = new LinkedList<>(); - for (String[] ticket : tickets) { - flights.putIfAbsent(ticket[0], new PriorityQueue<>()); - flights.get(ticket[0]).add(ticket[1]); + for (List ticket : tickets) { + flights.putIfAbsent(ticket.get(0), new PriorityQueue<>()); + flights.get(ticket.get(0)).add(ticket.get(1)); } dfs("JFK", flights, path); return path;