重新安排行程

题目

给你一份航线列表 tickets ,其中 tickets[i] = [fromi, toi] 表示飞机出发和降落的机场地点。请你对该行程进行重新规划排序。

所有这些机票都属于一个从 JFK(肯尼迪国际机场)出发的先生,所以该行程必须从 JFK 开始。如果存在多种有效的行程,请你按字典排序返回最小的行程组合。

例如,行程 ["JFK", "LGA"] 与 ["JFK", "LGB"] 相比就更小,排序更靠前。
假定所有机票至少存在一种合理的行程。且所有的机票 必须都用一次 且 只能用一次。

示例 1:

输入:tickets = [["MUC","LHR"],["JFK","MUC"],["SFO","SJC"],["LHR","SFO"]]
输出:["JFK","MUC","LHR","SFO","SJC"]
示例 2:

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

思路

  • 首先是构思如何 存储这个结构,采用哈希映射的方法,key为出发的机场,value为到达的机场
  • 字母的顺序表示可以使用有序哈希结构,也就是可以用第二个表的方式:unordered_map<string, map<string, int>> targets
    第一个string表示出发的机场,第二个map表示到达的机场,最后的数据类型int表示次数,这样存储的好处是第一次找到完整方案的时候就可以直接返回了。
  • 怎么循环这个targets也是需要注意的,表示内部的一个对称结构,可以使用pair<String, int>来表示

代码


class Solution {
private:
// unordered_map<出发机场, map<到达机场, 航班次数>> targets
unordered_map<string, map<string, int>> targets;
vector<string> result;
vector<string> path;
bool backtracking(int ticketNum) {
    if (path.size() == ticketNum + 1) {//每张票都要循环选取一遍,所以是ticketNum+1
        result = path;
        return true;
    }
    for (pair<const string, int>& target : targets[path[path.size() - 1]]) {
        if (target.second > 0 ) { // 记录到达机场是否飞过了
            path.push_back(target.first);
            target.second--;
            if(backtracking(ticketNum)) {
                return true;
            }
            path.pop_back();
            target.second++;
        }
    }
    return false;
}
public:
    vector<string> findItinerary(vector<vector<string>>& tickets) {
        targets.clear();
        for (const vector<string>& vec : tickets) {
            targets[vec[0]][vec[1]]++; // 记录映射关系
        }
        path.push_back("JFK"); // 起始机场
        backtracking(tickets.size());
        return result;
    }
};

posted @ 2023-04-19 11:46  铜锣湾陈昊男  阅读(3)  评论(0)    收藏  举报