给你一份航线列表 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<String, TreeSet> 存储tickets key表示出发地点 TreeSet表示终点 并且TreeSet自动会按字典进行排序 因为每次遍历直接取的是最佳路线 所以直接取TreeSet中的第一个即可
思路1错了 按照思路1的话每次可以选择多个终点的时候 如果只默认选第一个 有可能进入的地方没有可以去的下一个地方
思路2:
所以应该每次有多个可以去的地方的时候 要用递归每个地方都尝试一下 然后如果下个地方没有机票 就中断递归
然后当所有递归完成了 可能有多个结果 因为我们用的是TreeSet 所以肯定默认选最优的那个
所以当找到了结果的时候直接终止递归返回这个结果就行 另外因为java对对象的参数传递是引用传递
所以递归的时候需要不断回溯我们的HashMap<String,TreeSet>
思路2错了 TreeSet不能保存重复值 [[“JFK”,“SFO”],[“JFK”,“SFO”],[“SFO”,“JFK”]]用例失败
思路3(通过):
应该用HashMap<String, TreeMap<String,Integer>> 这样即可以排序String 也可以记录同一个机票的剩余票数
public static void dfs(List<String> sum, Boolean[] haveFind, String startStation, HashMap<String, TreeMap<String, Integer>> map, int sumStation, int nowStation) {
if (nowStation == sumStation) {// 已经找到结果了 并且第一个找到的就是最佳结果 结束所有递归
haveFind[0] = true;
return;
}
if (map.get(startStation) == null) return;//找不到始发站为startStation的票 这个递归失败
TreeMap<String, Integer> treeMap = map.get(startStation);
TreeSet<String> 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<String> findItinerary(List<List<String>> tickets) {
HashMap<String, TreeMap<String, Integer>> map = new HashMap<>();
for (List<String> ticket : tickets) {
if (map.containsKey(ticket.get(0))) {
TreeMap<String, Integer> 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<String, Integer> treeMap = new TreeMap<>();
treeMap.put(ticket.get(1), 1);
map.put(ticket.get(0), treeMap);
}
}
List<String> sum = new ArrayList<>();
sum.add("JFK");
Boolean[] haveFind = new Boolean[1];
haveFind[0] = false;
dfs(sum, haveFind, "JFK", map, tickets.size(), 0);
return sum;
}