力扣332. 重新安排行程 Java dfs回溯

给你一份航线列表 tickets ，其中 tickets[i] = [fromi, toi] 表示飞机出发和降落的机场地点。请你对该行程进行重新规划排序。

所有这些机票都属于一个从 JFK（肯尼迪国际机场）出发的先生，所以该行程必须从 JFK 开始。如果存在多种有效的行程，请你按字典排序返回最小的行程组合。

例如，行程 [“JFK”, “LGA”] 与 [“JFK”, “LGB”] 相比就更小，排序更靠前。
假定所有机票至少存在一种合理的行程。且所有的机票 必须都用一次 且 只能用一次。

示例 1：

输入：tickets = [["MUC","LHR"],["JFK","MUC"],["SFO","SJC"],["LHR","SFO"]]
输出：["JFK","MUC","LHR","SFO","SJC"]

示例 2：

输入：tickets = [["JFK","SFO"],["JFK","ATL"],["SFO","ATL"],["ATL","JFK"],["ATL","SFO"]]
输出：["JFK","ATL","JFK","SFO","ATL","SFO"]
解释：另一种有效的行程是 ["JFK","SFO","ATL","JFK","ATL","SFO"] ，但是它字典排序更大更靠后。

提示：

1 <= tickets.length <= 300
tickets[i].length == 2
fromi.length == 3
toi.length == 3
fromi 和 toi 由大写英文字母组成
fromi != toi

解题思路:
思路1:用HashMap 存储tickets key表示出发地点 TreeSet表示终点 并且TreeSet自动会按字典进行排序 因为每次遍历直接取的是最佳路线 所以直接取TreeSet中的第一个即可

思路1错了 按照思路1的话每次可以选择多个终点的时候 如果只默认选第一个 有可能进入的地方没有可以去的下一个地方
思路2:
所以应该每次有多个可以去的地方的时候 要用递归每个地方都尝试一下 然后如果下个地方没有机票 就中断递归
然后当所有递归完成了 可能有多个结果 因为我们用的是TreeSet 所以肯定默认选最优的那个
所以当找到了结果的时候直接终止递归返回这个结果就行 另外因为java对对象的参数传递是引用传递
所以递归的时候需要不断回溯我们的HashMap

思路2错了 TreeSet不能保存重复值 [[“JFK”,“SFO”],[“JFK”,“SFO”],[“SFO”,“JFK”]]用例失败
思路3(通过）:
应该用HashMap> 这样即可以排序String 也可以记录同一个机票的剩余票数

public static void dfs(List sum, Boolean[] haveFind, String startStation, HashMap> map, int sumStation, int nowStation) {
    if (nowStation == sumStation) {// 已经找到结果了 并且第一个找到的就是最佳结果 结束所有递归
        haveFind[0] = true;
        return;
    }
    if (map.get(startStation) == null) return;//找不到始发站为startStation的票 这个递归失败

    TreeMap treeMap = map.get(startStation);
    TreeSet ss = new TreeSet<>(treeMap.keySet());//new一个keySet来遍历treeMap 防止遍历时因为修改treeMap报错
    for (String s : ss) {//遍历所有剩余的始发站为startStation的票
        sum.add(s);
        Integer left = treeMap.get(s) - 1;//表示这个出发和目的地相同的机票还有几张
        if (left == 0) treeMap.remove(s);
        else treeMap.put(s, left);
        map.put(startStation, treeMap);
        dfs(sum, haveFind, s, map, sumStation, nowStation + 1);

        if (haveFind[0]) return;//已经找到结果了 不用递归了

        //回溯
        sum.remove(sum.size() - 1);
        treeMap.put(s, left + 1);
        map.put(startStation, treeMap);
    }
}

public static List findItinerary(List> tickets) {
    HashMap> map = new HashMap<>();
    for (List ticket : tickets) {
        if (map.containsKey(ticket.get(0))) {
            TreeMap treeMap = map.get(ticket.get(0));
            if (treeMap.containsKey(ticket.get(1))) treeMap.put(ticket.get(1), treeMap.get(ticket.get(1)) + 1);
            else treeMap.put(ticket.get(1), 1);
            map.put(ticket.get(0), treeMap);
        } else {
            TreeMap treeMap = new TreeMap<>();
            treeMap.put(ticket.get(1), 1);
            map.put(ticket.get(0), treeMap);
        }
    }
    List sum = new ArrayList<>();
    sum.add("JFK");
    Boolean[] haveFind = new Boolean[1];
    haveFind[0] = false;
    dfs(sum, haveFind, "JFK", map, tickets.size(), 0);
    return sum;
}