Reconstruct Itinerary
该题是常规的图搜索相关的题目,使用dfs便可以解决,可惜自己还是折腾了好久。
先贴上我原先的代码:
#include <iostream>
#include <vector>
#include <set>
#include <string>
#include <map>
#include <unordered_map>
using namespace std;
class Solution {
public:
vector<string> findItinerary(vector<pair<string, string>> tickets) {
unordered_map<string, set<string>> m;
for(auto item : tickets){
m[item.first].insert(item.second);
}
vector<string> res;
set<pair<string, string>> visited;
int total = tickets.size()+1;
string place = "JFK";
res.push_back(place);
dfs(m, place, res, visited, total);
return res;
}
bool dfs(unordered_map<string, set<string>> &m, string place, vector<string> &res, set<pair<string, string>> &visited, int total){
if(res.size() == total){
return true;
}
for(auto item : m[place]){
if(visited.find({place, item}) == visited.end()){
visited.insert({place, item});
res.push_back(item);
if(dfs(m, item, res, visited, total))
return true;
visited.erase({place,item});
res.pop_back();
}
}
return false;
}
};
int main()
{
vector<pair<string, string>> tickets = {{"JFK","SFO"},{"JFK","ATL"},{"SFO","ATL"},{"ATL","JFK"},{"ATL","SFO"}};
Solution s;
auto res = s.findItinerary(tickets);
for(auto i : res){
cout<<i<<endl;
}
return 0;
}
折腾了半天结果复杂测试样例还是超时了。然后对比了网上其他人的解法,发现整体的思路是一样的,对一些细节进行改进便能够很好的提高代码的性能。
改进后的代码如下:
class Solution {
public:
vector<string> findItinerary(vector<pair<string, string>> tickets) {
unordered_map<string, map<string, int>> m;
//直接多添加一位来记录该飞机票是否被使用,而不是额外使用一个set来保存,
//能使用unorder_map的地方尽量不使用map,set也同理
for(const auto &item : tickets){ //改成引用
m[item.first][item.second]++; //注意此时map的用法
}
vector<string> res;
int total = tickets.size()+1;
string place = "JFK";
res.push_back(place);
dfs(m, place, res, total);
return res;
}
bool dfs(unordered_map<string, map<string, int>> &m, const string &place, vector<string> &res, const int &total){ //参数能引用的就引用,能const的就const
if(res.size() == total){
return true;
}
for(auto &item : m[place]){ //改成引用
if(item.second != 0){
item.second--;
res.push_back(item.first);
if(dfs(m, item.first, res, total))
return true;
item.second++;
res.pop_back();
}
}
return false;
}
};
做出以上的改进后,代码运行时间为40ms,beats 75%。细节还是很重要的。
这题虽然思路不是很复杂,但是有很多地方需要注意:
- 首先考察了对STL使用的熟练程度,这也是导致我写了很久的原因。
- 还有提高代码效率的细节,比如unordered_map和map的选择,unordered_set和set的选择,以及传参和遍历时引用的使用,还有const的使用