代码随想录算法训练营 Day 30 | 332.重新安排行程,51.N皇后,37.解数独

最新推荐文章于 2024-09-04 20:50:47 发布

丢硬币的御坂美琴

最新推荐文章于 2024-09-04 20:50:47 发布

阅读量55

点赞数

文章标签：算法 java c++

本文链接：https://blog.csdn.net/MisakaMirai/article/details/130516923

版权

332.重新安排行程

讲解链接：代码随想录-332.重新安排行程

注意函数返回值我用的是bool！

我们之前讲解回溯算法的时候，一般函数返回值都是void，这次为什么是bool呢？

因为我们只需要找到一个行程，就是在树形结构中唯一的一条通向叶子节点的路线，如图：

public List<String> findItinerary(List<List<String>> tickets) {
    List<String> res = new ArrayList<>();
    Collections.sort(tickets, (a1, a2) -> a1.get(1).compareTo(a2.get(1)));
    // 外部定义使用过的集合，是用来检查当前深度的，如果是迭代方法内创建的，则是用来检查当前层的。
    boolean[] used = new boolean[tickets.size()];
    List<String> paths = new ArrayList<>();
    paths.add("JFK");
    backTracking(tickets, used, paths, res);
    return res;
}

boolean backTracking(List<List<String>> tickets, boolean[] used, List<String> paths, List<String> res) {
    if (paths.size() == tickets.size() + 1) {
        res.addAll(paths);
        return true;
    }
    for (int i = 0; i < tickets.size(); i++) {
        if (!used[i] && paths.get(paths.size() - 1).equals(tickets.get(i).get(0))) {
            paths.add(tickets.get(i).get(1));
            used[i] = true;
            if (backTracking(tickets, used, paths, res)) {
                return true;
            }
            used[i] = false;
            paths.remove(paths.size() - 1);
        }
    }
    return false;
}