LeetCode 332 : Reconstruct Itinerary

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的使用