Leetcode_332. 重新安排行程_dfs欧拉路

332. 重新安排行程

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

说明:

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

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

思路   这题：从“JFK”出发，经过所有边1次且仅1次，就是求欧拉路，搜索{注意：输出路径字典序最小}

class Solution {
private:
static vector<vector<string>> T;
map<string,vector<int>> edge;
int cnt,num_e;
vector<string> res,tmp;
vector<bool> visit; //edge

void dfs(string cur){
    if(res.size()>0) return ;
    tmp.push_back(cur);

    if(cnt==num_e){
        res=tmp;
        return ;
    }
    for(auto x: edge[cur]){
        if(!visit[x]){
            visit[x]=true;
            cnt++;
            dfs(T[x][1]);
            cnt--;
            visit[x]=false;
        }
    }
    tmp.pop_back();
}
static bool cmp(int &i,int &j){
    return T[i][1]<T[j][1];
}
public:
    vector<string> findItinerary(vector<vector<string>>& tickets) {
    num_e=tickets.size();
    T=tickets;
    for(int i=0;i<num_e;i++){ //all edge
        string start=tickets[i][0];
        if(edge.find(start)==edge.end()){
            edge[start]=vector<int>();
        }
        edge[start].push_back(i);
    }
    visit=vector<bool>(num_e,0);

    for(auto &x:edge){// x pair 
        sort(x.second.begin(),x.second.end(),cmp);
    }
    dfs("JFK");
    return res;
}
};

import java.util.*;

class Solution {
    public List<String> findItinerary(List<List<String>> tickets) {
        // 因为逆序插入，所以用链表
        List<String> ans = new LinkedList<>();

        if (tickets == null || tickets.size() == 0)
            return ans;

        Map<String, PriorityQueue<String>> graph = new HashMap<>();

        for (List<String> pair : tickets) {
            // 因为涉及删除操作，我们用链表
            PriorityQueue<String> nbr = graph.computeIfAbsent(pair.get(0), k -> new PriorityQueue<>());
            nbr.add(pair.get(1));
        }

        // 按目的顶点排序

        visit(graph, "JFK", ans);

        return ans;
    }

    // DFS方式遍历图，当走到不能走为止，再将节点加入到答案
    private void visit(Map<String, PriorityQueue<String>> graph, String src, List<String> ans) {

        Stack<String> stack = new Stack<>();

        stack.push(src);

        while (!stack.isEmpty()) {
            PriorityQueue<String> nbr;

            while ((nbr = graph.get(stack.peek())) != null &&
                    nbr.size() > 0) {
                stack.push(nbr.poll());
            }
            ans.add(0, stack.pop());
        }
    }
}