leetcode 332. Reconstruct Itinerary

来源:互联网 发布:日韩男士帽子 知乎 编辑:程序博客网 时间:2024/06/07 05:08

原题链接

原题链接

解题思路

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

  1. 用Map存储图。存储后并将其列表按排序。
  2. 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;    }}
1 0
原创粉丝点击