leetcode 332. Reconstruct Itinerary

最新推荐文章于 2023-09-14 21:30:00 发布

jly0612

最新推荐文章于 2023-09-14 21:30:00 发布

阅读量405

点赞数 1

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.csdn.net/jly0612/article/details/51851750

版权

leetcode 专栏收录该内容

37 篇文章 0 订阅

订阅专栏

原题链接

解题思路

考察图以及深度搜索。dfs算法（也就是递归）注意一个是边界点，一个是状态恢复的过程。此题很明显当ans.size() == tickets.length+1结束。有些排在列表前面的顶点没有dest，就返回false，恢复原来的状态遍历下一个顶点。若

用Map存储图。存储后并将其列表按排序。
dfs算法遍历整个图。并且将遍历结果存储到list返回。

解题代码

public class Solution {
    Map<String,List<String>> map = new HashMap<>();
    int count;  
    public List<String> findItinerary(String[][] tickets) {
        count = tickets.length+1;
        for (String[] strs:tickets ) {
            List<String> dests = map.get(strs[0]);
            if(dests == null){
                dests = new ArrayList<>();
                dests.add(strs[1]);
                map.put(strs[0],dests);
            }else{
                dests.add(strs[1]);
            }
        }
        for(List<String> dests : map.values()){
            Collections.sort(dests);
        }
        List<String> ans = new ArrayList<>();
        ans.add("JFK");
        dfs(ans,"JFK");
        return ans;
    }
    public boolean dfs(List<String> ans,String src){
        if(ans.size() == count) return true;;
        List<String> dests = map.get(src);
        if(dests !=null && dests.size()>0){
            for (int i = 0;i<dests.size() ;i++ ) {
                String dest = dests.get(i);
                ans.add(dest);
                dests.remove(i);
                if(dfs(ans,dest)) return true;
                dests.add(i,dest);//恢复过程
                ans.remove(ans.size()-1);
            }
        }
        return false;
    }
}