刷怪日记-重新安排行程

332 重新安排行程

题目

解题思路

• 难点

  • 重复机场名字，行程变成一个圈，成为死循环
  • 多种解法，字母小靠前排在前面
  • 回溯法的终止条件如何选择

• 通过used[]数组记录航班使用情况，已使用记录为false，从而避免重复

• 将每个航班的终点站进行对比，小的排前面，从而实现字母小的先使用，比如说初始站相同，那么终点站字母小的就在前面先使用了

  tickets = [["JFK","SFO"],["JFK","ATL"],["SFO","ATL"],["ATL","JFK"],["ATL","SFO"]]
  

• 回溯遍历的过程中，遇到的机场个数，如果达到了（航班数量+1），那么就找到了一个行程，把所有航班串在一起了

具体代码

class Solution {
    private LinkedList<String> res;
    private LinkedList<String> path = new LinkedList<>();

    public List<String> findItinerary(List<List<String>> tickets) {
        Collections.sort(tickets, (a, b) -> a.get(1).compareTo(b.get(1))); //排序，相同初始站，终点站字母小的排前面
        path.add("JFK"); // 初始起始站
        boolean[] used = new boolean[tickets.size()]; // 初始化false，代表均未使用
        backTracking((ArrayList) tickets, used);
        return res;
    }

    public boolean backTracking(ArrayList<List<String>> tickets, boolean[] used) {
        if (path.size() == tickets.size() + 1) { // 机场个数 = 航班数量+1，代表找到一个新的行程
            res = new LinkedList(path);
            return true;
        }

        for (int i = 0; i < tickets.size(); i++) {
            if (!used[i] && tickets.get(i).get(0).equals(path.getLast())) { // 未使用，且该航程初始站与行程的最后一个元素相同
                path.add(tickets.get(i).get(1));
                used[i] = true;

                if (backTracking(tickets, used)) { // 如果找到一个行程就一直true结束，因为行程唯一
                    return true;
                }

                used[i] = false;
                path.removeLast();
            }
        }
        return false;
    }
}