LeetCode332-重新安排行程

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

说明:

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

示例 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"]。但是它自然排序更大更靠后。

一、思路

（一）深度优先搜索

这是一个图的问题，上述所说的“地点”代表顶点，“行程”代表边，由此建立图。
然后题目要求，字母排序最小的行程组合，那我们不妨将地点映射为数字，以此表示，最后则是DFS找出合理的行程。

总结一下，算法流程：

• （1）将地点按照字典序顺序，映射为数字
• （2）设置一个数组用于保存每个顶点的出边所指向的顶点
• （3）DFS

C++代码：

class Solution {
public:
    vector<string> ans;
    map<string, int> hash;
    vector<string> list;
    int len1, len2;
    vector<string> findItinerary(vector<vector<string>>& tickets) {
        for(auto tic : tickets){
            list.push_back(tic[0]);
            list.push_back(tic[1]);
        }
        sort(list.begin(), list.end());
        int id = 0;
        for(int i=0; i < list.size(); i++){
            if(hash.count(list[i]) == 0){
                hash.insert(pair<string, int>(list[i], id));
                id++;
            }
            else{
                list.erase(list.begin() + i);
                i--;
            }
        }
        // 建立好了：机场名称-->数字的映射
        len1 = tickets.size();
        len2 = list.size();
        vector<int> out_degree(id, 0);
        vector<vector<int>> out(id);
        for(auto tic : tickets){
            out_degree[hash[tic[0]]]++;
            out[hash[tic[0]]].push_back(hash[tic[1]]);
        }
        for(int i=0; i < id; i++)
            sort(out[i].begin(), out[i].end());
        // 准备就绪
        int begin = hash["JFK"];
        dfs(out, begin, 0);
        return ans;
    }

    bool dfs(vector<vector<int>>& out, int begin, int num){
        ans.push_back(list[begin]);
        if(num == len1)
            return true;
        bool fl = false;;
        for(int i=0; i < out[begin].size(); i++){
            int temp = out[begin][i];
            out[begin].erase(out[begin].begin() + i);
            fl = dfs(out, temp, num + 1);
            if(fl)
                return true;
            else{
                ans.pop_back();
                out[begin].insert(out[begin].begin() + i, temp);
            }
        }
        return false;
    }
};