给定一个机票的字符串二维数组 [from, to],子数组中的两个成员分别表示飞机出发和降落的机场地点,对该行程进行重新规划排序。所有这些机票都属于一个从 JFK(肯尼迪国际机场)出发的先生,所以该行程必须从 JFK 开始,假定所有机票至少存在一种合理的行程。且所有的机票 必须都用一次 且 只能用一次。示例如下:
题目需要解决的几个问题:
(1)一个行程中,如果航班处理不好容易变成一个圈,进行死循环
(2)存在多个路径,字母排序在前的排在前面
(3)需要遍历一个机场对应的所有机场
解决办法如下:
一个机场对应几个机场,机场之间要考字母顺序进行排序,一对多可以使用unordered_map,多个机场有序可以使用multimap,所以存放的映射关系可以是unordered_map<string,mutiset<string>> targets含义为:unordered_map<出发机场, 到达机场的集合> targets;
或者为unordered_map<string,mutimap<string,int>> targets含义为:unordered_map<出发机场, map<到达机场, 航班次数>> targets。
由于出发机场和到达机场是会重复的,所以搜索过程中要不断删除multiset里面的元素,所以使用第二种映射关系对航班次数进行操作会更加容易。
回溯三部曲:
(1)递归参数和返回值:返回类型为bool型即可,因为只需要找到一个行程。
(2)确定终止条件:当机场个数达到了航班数量加一就找到一个行程可以返回。
(3)单层搜索逻辑。
实现代码如下:
//重新安排行程
//unordered_map<出发机场,map<到达机场,航班次数>> targets
unordered_map<string,map<string,int>> targets;
bool backtracking(vector<string>& result, int ticketNum) {
//确定终止条件
if(result.size() == ticketNum +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(result, ticketNum)) return true;
result.pop_back();
target.second++;
}
}
return false;
}
vector<string> findItinerary(vector<vector<string>>& tickets) {
targets.clear();
vector<string> result;
//进行初始化
for(const vector<string>& vec : tickets) {
targets[vec[0]][vec[1]]++; //记录映射关系
}
int ticketNum = tickets.size();
result.push_back("JFK"); //起始机场
backtracking(result, ticketNum);
return result;
}