前言
回溯章节第13篇。
记录 八十九【332.重新安排行程】
一、题目阅读
给你一份航线列表 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"] ，但是它字典排序更大更靠后。
提示：
1 <= tickets.length <= 300
tickets[i].length == 2
fromi.length == 3
toi.length == 3
from i 和 to i 由大写英文字母组成
from i != to i
二、尝试实现
2.1分析题目，确定方法
- 理解题目：
- 给的参数是tickets 数组，其中每一个元素代表一张机票，该机票表现形式是[fromi, toi] ，表示出发地和终点；如下图：
- 题目要求把tickets都串联起来，从JFK的机票开始，形成一条航线。比如示例一：
- 理解题目含义之后：
- 明显看出来这不是贪心或者动态规划类的题目。没有办法从前一个状态推到下一个状态。
- tickets集合可以将节点构成一个有向图，使用图论中的搜索方式，找到一条路径如何？应该可以。但是第一步需要先把tickets变成图的表示方法：邻接表。
- 根据记录 八十二【图论理论基础及深度优先搜索算法】和记录 八十五【图论的广度优先搜索理论基础】，广度优先搜索应用在两点之间最短路径查找多，本题目应该选用深度优先搜索，因为是找一条路径，当找不通时，回溯更改方向继续搜索。如果搜到路径（并且排序最小），应该返回。
- 那么：本题方法——深搜+回溯。
- 用到回溯算法：参考为回溯新开了一个章节。那么相当于从tickets集合中顺序选择机票，串联起来，找一个排列。（另一个角度，所以不需要把tickets转换成邻接表）
2.2 回溯思路【排列角度】
- 题目只返回一条行程，按字典排序最小的行程。那么，先把tickets进行从小到大排序。第一个问题：如何解决排序？
-
现有的sort函数没有办法对tickets[i]：vector< string> 形式排序，所以要自己实现比较函数。如下：
1. static修饰：如果调用类的非静态成员函数要创建对象。为了力扣的核心代码形式，所以加static。 2. 先比较from出发地。如果from一样，比较end终点。从小到大排列。 static bool compare(const vector<string>& ticket_a,const vector<string>& ticket_b){ if(ticket_a[0] == ticket_b[0]){ return ticket_a[1] < ticket_b[1]; }else{ return ticket_a[0] < ticket_b[0]; } } -
排序之后，找到的第一条路径肯定是字典排序后最小的。 只搜集第一个叶子结点。
- 递归搜索：回溯的过程。
- 确定递归的返回值：用一个成员vector< string> temp;记录结果。
- 确定递归的参数：
- vector< bool>& used：表明这个tickets[i]有没有被使用过。题目说所有的tickets[i]都要被使用且使用1次，所以used作用在此；
- string from ：代表下一个机票的出发地 。上一个ticket的降落地 = 下一个ticket的出发地；
- int& usenum：代表使用过的tickets总数。控制终止条件。usenum == tickets.size()时终止；
- const vector<vector< string>>& tickets ：输入的tickets数组。
- 确定递归的终止条件：
- 如果从temp.size来判断终止不太方便，所以用一个变量usenum表示现在使用了多少个ticket，如果usenum == tickets.size()代表使用了所有的tickets，那么应该return。
- 确定递归的逻辑：
- 在主函数中先sort tickets数组，结果相当于求一个排列，所以每次for循环从0开始到tickets.size()。
- 重复使用处理：if(used[i]) continue;如果标记该tickets[i]使用过，那么continue；
- 地点不连续处理：if(tickets[i][0] != from) continue;如果出发地不是上一层传递的from，那么continue；
- 找到的第一个没使用过且from出发地符合的tickets[i]一定是最小路径的一员。（提醒排过序了）
- tickets处理：把该tickets[i]的目的地放入temp；used = true；usenum++；
- 递到下一层：from赋值上一步的tickets[i]的目的地（toi）;
- 回溯：把temp放入的目的地pop；used = false；usenum - -；
- 细节处理：本题只返回一条排序最小的路径，所以只搜集一个叶子结点。
- 设置一个全局的标志：bool ok = false；
- 在终止条件中，ok = true；
- 在for循环中，回溯之前先判断if(ok) ，那么break;
- 该回溯实现的树形结构，以示例一为例：
2.3 代码实现【回溯算法+排列角度】
class Solution {
public:
vector<string> temp;//中间结果
bool ok= false;
static bool compare(const vector<string>& ticket_a,const vector<string>& ticket_b){
if(ticket_a[0] == ticket_b[0]){
return ticket_a[1] < ticket_b[1];
}else{
return ticket_a[0] < ticket_b[0];
}
}
void backtracking(vector<bool>& used,string from,int& usenum,const vector<vector<string>>& tickets){
//终止条件，是否所有机票都被使用过
if(usenum == tickets.size()){
ok = true;
return;
}
for(int i = 0;i < tickets.size();i++){
if(tickets[i][0] != from) continue;
if(used[i]) continue;
temp.push_back(tickets[i][1]);//把目的地放入
used[i] = true;//该机票使用过
usenum++;
backtracking(used,tickets[i][1],usenum,tickets);//递到下一层
if(ok) break;//只返回最小的行程，所以另一终止条件.只搜集一个叶子节点。
temp.pop_back();
used[i] = false;
usenum--;
}
}
vector<string> findItinerary(vector<vector<string>>& tickets) {
sort(tickets.begin(),tickets.end(),compare);//把所有机票从小到大排列
temp.clear();
temp.push_back("JFK");
vector<bool> used(tickets.size(),false);
int usenum = 0;
backtracking(used,"JFK",usenum,tickets);
return temp;
}
};
问题：提交后超时，无法通过；
- 分析时间复杂度：
- 首先sort，排序方式快排，时间复杂度O(nlogn)；
- 在进入递归：递归次数最差情况需要找到排列树形结构中的最后一个叶子节点。所以时间复杂度是O(n!)；
- 那么整体的时间复杂度：O(n* n! * logn)；
三、参考学习
如何解决超时的问题？
3.1 学习思路
- 深搜离不开回溯。这道题就是深度优先搜索，深搜中必须用到回溯。
- 问题一：如何避开死循环？以参考给的示例：如果处理不好会在“JFK”和“NRT”之间来回跳跃。
- 尝试实现没有出现这个问题：使用used数组记录是否使用过，且使用1次，并不会出现这种情况。进入死循环，是因为把某个机票重复使用多次，或者有重复的机票。
- 问题二：处理映射关系。一个出发地可能对应有多个目的地。
解决：在unordered_map中使得出发机场没有重复的，且无需排序，底层实现是哈希表；map不允许重复，有序，底层实现是红黑树（二叉搜索树）；
unordered_map<string, map<string, int>> targets：unordered_map<出发机场, map<到达机场, 航班次数>> targets
航班次数用来解决：tickets中有重复的机票。如果“航班次数”大于零，说明目的地还可以到达，如果“航班次数”等于零说明目的地无法到达。
map<到达机场, 航班次数>完成2.3代码中的排序操作。
- 递归回溯实现：
- 递归的参数：
- unordered_map<string, map<string, int>> targets：在函数之前先解决映射关系；
- vector < string>& result：存放路径结果；相当于2.3代码的temp；
- int ticketNum。终止条件使用，表示航班总数（终止条件会用上）= tickets数组的总数。
- 递归的返回值：bool
- 2.2中的细节处理：只搜集一个叶子节点，个人用bool ok做了标记；在for循环中，回溯之前先判断if(ok) break;
- 参考使用：返回值bool。
- 递归终止条件：
if (result.size() == ticketNum + 1) { return true; } - 递归函数的逻辑：
- 遍历result[result.size()-1]（相当于2.3代码中的from参数的作用）所映射的map；
- 先判断**“航班次数”** 是否大于0，还有无机会到达该目的地；
- 处理航班：如果可以，把**“到达机场”** 放入result中，再“航班次数”++；
- 递到下一层，根据下一层的返回值，是否直接return true；相当于2.3代码中的if(ok) break;
- 回溯操作：把**“到达机场”** 从result中pop，再“航班次数”- -；
- 递归的参数：
3.2 代码实现【回溯+航班映射】
class Solution {
public:
unordered_map<string,map<string,int>> targets;
bool backtracking(int ticketSum,vector<string>& result){
if(result.size() == ticketSum+1){
return true;
}
for(pair<const string,int>& target:targets[result[result.size()-1]]){
if(target.second > 0){
result.push_back(target.first);
target.second--;
if(backtracking(ticketSum,result)) return true;
result.pop_back();
target.second++;
}
}
return false;
}
vector<string> findItinerary(vector<vector<string>>& tickets) {
targets.clear();
for(int i = 0;i < tickets.size();i++){
targets[tickets[i][0]][tickets[i][1]]++;
}
vector<string> result;
result.push_back("JFK");
backtracking(tickets.size(),result);
return result;
}
};
注意点：
-
主函数中result要先放入起点“JFK”；
-
在使用rangefor遍历from出发地映射的map时，target必须是引用，因为target.second做了修改，需要传递到下一层；
-
pair<const string,int>类型必须加const，因为target必须是引用，就必须在 string 前面加上 const，因为map中的key 是不可修改的。或者直接auto让编译器自动识别类型。
for(auto& target:targets[result[result.size()-1]])
3.3 分析时间复杂度
对比2.3和3.2 两个代码实现：深搜和回溯的思路一致，但是容器的使用和处理上不一样导致一个出现超时的用例，一个没有超时风险。
- 完成映射的时间复杂度是O(n)；
- 进入到递归中。使用unordered_map找到from映射的map，底层是哈希表，查找O(1)常数级；再考虑递归次数：从排列角度看结果是获得一个排列，搜索顶多O(n!)；
- 那么3.2的时间复杂度是O(n * n!)；
- 因为使用映射容器，使得在查找from出发地时，用unordered_map减少了时间；递归中的for循环时间减少，并且排序时间也减少，两个地方使得3.2的代码实现比2.3更好。
总结
（欢迎指正，转载标明出处）
扫一扫
1106
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
