Leetcode刷题笔记 332. 重新安排行程

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

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

说明：

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

解法

 1. 因为本题保证至少存在一种合理的路径，也就告诉了我们，这张图是一个欧拉图或者半欧拉图。我们只需要输出这条欧拉通路的路径即可。
 2.当图为
        s
      //  \\
      b  =  c
    有两种走法 s-b-c 和 s-c-b
    我们选择字典序小的s-b-c，可以用unordered_map<string, priority_queue<string, vector<string>, greater<string>>>
    从起点出发，进行深度优先搜索，找连接的字典序最小的节点，加入路径，把这条边弹出，这个节点继续搜索，
    返回条件：所有路径都走过了，返回true，否则false
    当false时，路径中弹出节点，把这条边加入。
3. 但是当图为
         a
        /
       s
     //
    b  =  c
  我们先走a，发现a行不通，回溯，因为优先队列，我们每次取的是a，出错
  我们注意到只有那个入度与出度差为1的节点（a）会导致死胡同。而该节点必然是最后一个遍历到的节点。
  改变加入路径的规则，当遍历完一个节点所连的所有节点后，我们才将该节点入栈（即逆序入栈）。
  s - a、b
  b - c、s
  c - b
 路径为 a s b c b s
 
 

代码

#include <stdio.h>
#include <iostream>
#include <vector>
#include <cstring>
#include <queue>
#include <unordered_map>
using namespace std;
class Solution {
public:
    unordered_map<string, priority_queue<string, vector<string>, greater<string>>> dic;
    vector<string> ans;
    void dfs(const string& curr) {
        while (dic.count(curr) && dic[curr].size() > 0) {
            string tmp = dic[curr].top();
            dic[curr].pop();
            dfs(tmp);
        }
        ans.push_back(curr);
    }
    vector<string> findItinerary(vector<vector<string>>& tickets) {
        int n = tickets.size();
        for(int i=0;i<n;i++){
            dic[tickets[i][0]].push(tickets[i][1]);
        }
        dfs("JFK");
        reverse(ans.begin(), ans.end());
        return ans;
    }
};
int main()
{
    vector<vector<string>> tickets;
//    vector<string> s1;s1.push_back("JFK");s1.push_back("SFO");
//    vector<string> s2;s2.push_back("JFK");s2.push_back("ATL");
//    vector<string> s3;s3.push_back("SFO");s3.push_back("ATL");
//    vector<string> s4;s4.push_back("ATL");s4.push_back("JFK");
//    vector<string> s5;s5.push_back("ATL");s5.push_back("SFO");
    vector<string> s1;s1.push_back("JFK");s1.push_back("KUL");
    vector<string> s2;s2.push_back("JFK");s2.push_back("NRT");
    vector<string> s3;s3.push_back("NRT");s3.push_back("JFK");
    tickets.push_back(s1);tickets.push_back(s2);
    tickets.push_back(s3);
   // tickets.push_back(s4);tickets.push_back(s5);
    Solution s;
    vector<string> ans = s.findItinerary(tickets);
    for (int i=0;i<ans.size();i++)
        cout<<ans[i]<<endl;
    return 0;
}

今天也是爱zz的一天哦！