332. 重新安排行程

重新安排行程

给定一个机票的字符串二维数组 [from, to]，子数组中的两个成员分别表示飞机出发和降落的机场地点，对该行程进行重新规划排序。所有这些机票都属于一个从JFK（肯尼迪国际机场）出发的先生，所以该行程必须从 JFK 出发。

说明:

1. 如果存在多种有效的行程，你可以按字符自然排序返回最小的行程组合。例如，行程 [“JFK”, “LGA”] 与 [“JFK”, “LGB”] 相比就更小，排序更靠前
2. 所有的机场都用三个大写字母表示（机场代码）。
3. 假定所有机票至少存在一种合理的行程。

示例 1:

输入: [["MUC", "LHR"], ["JFK", "MUC"], ["SFO", "SJC"], ["LHR", "SFO"]]
输出: ["JFK", "MUC", "LHR", "SFO", "SJC"]

示例 2:

输入: [["JFK","SFO"],["JFK","ATL"],["SFO","ATL"],["ATL","JFK"],["ATL","SFO"]]
输出: ["JFK","ATL","JFK","SFO","ATL","SFO"]
解释: 另一种有效的行程是 ["JFK","SFO","ATL","JFK","ATL","SFO"]。但是它自然排序更大更靠后。

思路+代码+注释：

class Node{
        private String val;
        private TreeSet<String> desCities=new TreeSet<String>();

        public Node(String val) {
            this.val = val;
        }

        public TreeSet<String> getDesCities() {
            return desCities;
        }
    }
    class Tu{
        private List<Node> nodes=new ArrayList<>();

        public void add(Node node)
        {
            nodes.add(node);
        }
        public boolean contains(String val)
        {
            for (Node n:nodes
                 ) {
                if (n.val==val)
                {
                    return true;
                }
            }
            return false;
        }
        public Node getNode(String val)
        {
            for (Node n:nodes
                 ) {
                if (n.val==val)
                {
                    return n;
                }
            }
            return null;
        }
    }



    public List<String> findItinerary(String[][] tickets) {
            /*
            思路：首先使用邻接表表示图这种结构，机票相当于边，将机票的元素当作顶点添加到图中，构建图完成
            从JFK元素开始，获取该元素可达的下一个顶点，将该顶点从图中删除，然后递归从该顶点开始，获取该顶点可达的下一个顶点，依次类推，直到
            顶点没有下一个可达顶点，最后将每次开始找的顶点添加到res中，再将res翻转就得到结果
             */
            Tu tu=new Tu();
        for (int i = 0; i < tickets.length; i++) {
            String[] ticket=tickets[i];
            String source=ticket[0];
            String des=ticket[1];
            if (tu.contains(source))
            {
                Node node=tu.getNode(source);
                TreeSet<String> treeSet=node.getDesCities();
                treeSet.add(des);
            }else {
                Node node=new Node(source);
                TreeSet<String> treeSet=node.getDesCities();
                treeSet.add(des);
                tu.add(node);
            }
        }
        List<String> res=new ArrayList<>();
        findCiti(tu,res,"JFK");
        List<String> resList=new ArrayList<>(res.size());
        for (int i = res.size()-1; i >-1 ; i--) {
            resList.add(res.get(i));
        }
            return resList;
    }

    private void findCiti(Tu tu,List<String> res,String source)
    {
        Node node=tu.getNode(source);
        if (node!=null)
        {
            TreeSet<String> destinations=node.getDesCities();
            if (destinations.size()>0)
            {
                String des=destinations.first();
                destinations.remove(des);
                findCiti(tu,res,des);
            }
        }
        res.add(source);
    }