leetcode困难332重新安行程

Posted 2021-12-09 qq_40707462

思路：回溯
1、类似全排列，每一次 for 都全部遍历，用 visited 数组记录是否使用过
2、题目只要求一种答案，即有答案了就return，为了使 dfs 函数没有返回值方便代码，设置全局 flag，flag 为 true 就停止。
3、要找字典序最小的，可以提前将数组排列好，这样 dfs 到的第一个就是字典序最小的

class Solution {
    List<String> res= new ArrayList<>();
    List<String> path= new ArrayList<>();
    boolean flag=false;
    boolean[] visited;
    public List<String> findItinerary(List<List<String>> tickets) {
        visited=new boolean[tickets.size()];
        // 将票组按照目的地的字典顺序从小到大排序
        tickets.sort((o1, o2) -> o1.get(1).compareTo(o2.get(1))); 
        dfs(tickets,0,"JFK");
        res.add(0, "JFK"); // 将出发地添加到结果当中
        return res; // 返回结果
    }
    public void dfs(List<List<String>> tickets,int level,String lastPop){
        // 如果第一次找到了答案，则这个答案就是最优的，因为已经排序好了
        if (level == tickets.size()) { 
            res = new ArrayList<>(path); // 将结果赋值
            flag = true; // 将flag设置为true
            return;
        }
        for (int i = 0; i < tickets.size(); i++) { // 每次遍历全部
            List<String> ticket = tickets.get(i);
            if (ticket.get(0).equals(lastPop) && !visited[i] && !flag) {
                path.add(ticket.get(1));
                visited[i] = true;
                dfs(tickets, level + 1, ticket.get(1));
                visited[i] = false;
                path.remove(path.size() - 1);
            }
        }
    }
}