Leetcode 332. 重新安排行程 C++

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

题解

dfs深搜回溯
建立邻接表，同时记录那种航班已经使用过了，以避免重复使用。详细过程见代码

代码

	vector<string> ans;
    unordered_map<string,vector<string>> ticket;		//邻接表
    unordered_map<string,vector<int>> use;				//记录那个航班使用过
    bool dfs(string& now,int begin,int n){
        if(begin == n){
            return true;
        }else{
            int size = ticket[now].size();
            for(int i=0; i<size; i++){
                if(!use[now][i]){
                    ans.push_back(ticket[now][i]);
                    use[now][i] = 1;
                    if(dfs(ticket[now][i],begin+1,n))   return true;
                    ans.pop_back();
                    use[now][i] = 0;
                }
            }
        }
        return false;
    }
    vector<string> findItinerary(vector<vector<string>>& tickets) {
        int n = tickets.size();
        for(int i=0; i<n; i++){
            int j,n = ticket[tickets[i][0]].size();
            for(j=0; j<n; j++)		//插入排序的方式，保证有序
                if(ticket[tickets[i][0]][j] >= tickets[i][1])   break;		
            ticket[tickets[i][0]].insert(ticket[tickets[i][0]].begin()+j,tickets[i][1]);
            use[tickets[i][0]].push_back(0);
        }
        string beginC = "JFK";
        ans.push_back(beginC);		//起始位置
        dfs(beginC,0,n);
        return ans;    
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/reconstruct-itinerary
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。