原题地址:https://leetcode-cn.com/problems/reconstruct-itinerary/submissions/
题目描述:
给定一个机票的字符串二维数组 [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"]。但是它自然排序更大更靠后。
解题方案:
不会做。。。
找的代码:
class Solution {
public:
struct cmp
{
bool operator()(const string& a, const string& b) {
return a > b;
}
};
vector<string> findItinerary(vector<pair<string, string>> tickets) {
map<string, int> m;
vector<priority_queue<string, vector<string>, cmp > > queues;
for (auto p : tickets) {
if (m.find(p.first) == m.end()) {
priority_queue<string, vector<string>, cmp> newQueue;
newQueue.push(p.second);
queues.push_back(newQueue);
m.insert(make_pair(p.first, queues.size() - 1));
}
else {
queues[m[p.first]].push(p.second);
}
}
vector<string> ans;
stack<string> visitedPlaces;
visitedPlaces.push("JFK");
while (!visitedPlaces.empty()) {
string current = visitedPlaces.top();
if (m.find(current) == m.end() || queues[m[current]].size() == 0) {
ans.push_back(current);
visitedPlaces.pop();
}
else {
visitedPlaces.push(queues[m[current]].top());
queues[m[current]].pop();
}
}
reverse(ans.begin(), ans.end());
return ans;
}
};
以后再好好想想吧~