问题描述 :
给定一个机票的字符串二维数组 [from, to],子数组中的两个成员分别表示飞机出发和降落的机场地点,对该行程进行重新规划排序。所有这些机票都属于一个从 JFK(肯尼迪国际机场)出发的先生,所以该行程必须从 JFK 开始。
说明:
如果存在多种有效的行程,你可以按字符自然排序返回最小的行程组合。例如,行程 [“JFK”, “LGA”] 与 [“JFK”, “LGB”] 相比就更小,排序更靠前
所有的机场都用三个大写字母表示(机场代码)。
假定所有机票至少存在一种合理的行程。
示例 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”]。但是它自然排序更大更靠后。
可使用以下main函数:
int main()
{
vector<vector<string>> tickets;
int m;
cin>>m;
string a1,a2;
for(int i=0; i<m; i++)
{
vector<string> aTicket;
cin>>a1>>a2;
aTicket.push_back(a1);
aTicket.push_back(a2);
tickets.push_back(aTicket);
}
vector<string> res=Solution().findItinerary(tickets);
for(int i=0; i<res.size(); i++)
{
if (i>0)
cout<<" ";
cout<<res[i];
}
return 0;
}
输入说明 :
首先输入机票张数m,
然后输入m行,每行两个字符串,表示机场的代码,中间以空格分隔。
输出说明 :
输出结果
输入范例:
5
JFK SFO
JFK ATL
SFO ATL
ATL JFK
ATL SFO
输出范例 :
JFK ATL JFK SFO ATL SFO
void dfs(string from, map<string, multiset<string>> &from_to, vector<string> &ans) {
while (from_to[from].size() != 0) {
string next = *from_to[from].begin();
from_to[from].erase(from_to[from].begin());//可能存在相同的地点,所以只能erase迭代器
dfs(next, from_to, ans);
}
ans.push_back(from);//一定倒着存,将孤立节点放在最后访问。题目保证一定会有路径
}
//欧拉路径
vector<string> findItinerary(vector<vector<string>> &tickets) {
map<string, multiset<string>> from_to;//multiset相当于小根堆 自动从大到小排序而且允许重复
for (int i=0;i<tickets.size();i++){
from_to[tickets[i][0]].insert(tickets[i][1]);
}
vector<string> ans;
dfs("JFK", from_to, ans);
reverse(ans.begin(),ans.end());
return ans;//逆序输出
}