【发布时间】:2018-02-20 00:28:36
【问题描述】:
我正在解决 Leetcode 上的一个问题:https://leetcode.com/problems/reconstruct-itinerary/description/。问题是:
Given a list of airline tickets represented by pairs of departure and
arrival airports [from, to], reconstruct the itinerary in order. All of
the tickets belong to a man who departs from JFK. Thus, the itinerary
must begin with JFK.
例如,如果tickets = [["MUC", "LHR"], ["JFK", "MUC"], ["SFO", "SJC"], ["LHR", "SFO"]],则输出应为:["JFK", "MUC", "LHR", "SFO", "SJC"]。
我编写了以下代码,这(可以理解)在输入 [["JFK","KUL"],["JFK","NRT"],["NRT","JFK"]] 上中断,因为根据我的代码,节点“NRT”仍未被访问:
class Solution {
public:
vector<string> findItinerary(vector<pair<string, string>> tickets) {
if(tickets.empty()) return vector<string>();
vector<string> result;
unordered_map<string, multiset<string>> itinerary;
for(auto& each : tickets)
itinerary[each.first].insert(each.second);
stack<string> myStack;
myStack.push("JFK");
while(!myStack.empty()) {
string topVal=myStack.top();
result.push_back(topVal);
myStack.pop();
if(!itinerary[topVal].empty()) {
myStack.push(*itinerary[topVal].begin());
itinerary[topVal].erase(itinerary[topVal].begin());
}
}
return result;
}
};
为了克服这个问题,其中一个被赞成的解决方案提出了这个小改动:
class Solution {
public:
vector<string> findItinerary(vector<pair<string, string>> tickets) {
if(tickets.empty()) return vector<string>();
vector<string> result;
unordered_map<string, multiset<string>> itinerary;
for(auto& each : tickets)
itinerary[each.first].insert(each.second);
stack<string> myStack;
myStack.push("JFK");
while(!myStack.empty()) {
string topVal=myStack.top();
if(itinerary[topVal].empty()) { //--->this if condition
result.push_back(topVal);
myStack.pop();
}
else {
myStack.push(*itinerary[topVal].begin());
itinerary[topVal].erase(itinerary[topVal].begin());
}
}
reverse(result.begin(), result.end());
return result;
}
};
现在,我使用示例 [["JFK","KUL"],["JFK","NRT"],["NRT","JFK"]] 编写此代码,并查看它如何以相反的方式将值插入到 result 向量中;但我无法理解 if 条件背后的直觉:
如何只在集合为空时才从栈中弹出,确保这个测试用例得到照顾?
【问题讨论】:
标签: c++ algorithm graph graph-traversal