Reconstruct Itinerary

最新推荐文章于 2019-08-07 08:52:40 发布

原创最新推荐文章于 2019-08-07 08:52:40 发布 · 304 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c语言

LeetCode 专栏收录该内容

109 篇文章

订阅专栏

本文介绍了一种使用递归回溯的方法来解决行程重构问题的算法。该算法通过深度优先搜索（DFS）的方式遍历所有可能的行程路径，并最终返回一条从起点出发能够访问所有地点并回到起点的路径。算法首先构建了一个图数据结构，然后从指定的起点开始进行深度优先搜索，直到找到解决方案。

c++

class Solution {
public:
    vector<string> findItinerary(vector<pair<string, string>> tickets) {
        if (tickets.empty()) return vector<string>();
        unordered_map<string, multiset<string>> graph;
        for (auto &v : tickets) {
            graph[v.first].insert(v.second);
        }
        stack<string> dfs;
        dfs.push("JFK");
        vector<string> res;
        while (!dfs.empty()){
            string cur_node = dfs.top();
            if (graph[cur_node].empty()) {
                res.push_back(cur_node);
                dfs.pop();
            }
            else {
                dfs.push(*graph[cur_node].begin());
                graph[cur_node].erase(graph[cur_node].begin());
            }
        }
        reverse(res.begin(), res.end());
        return res;
    }
};

python

class Solution(object):
    def findItinerary(self, tickets):
        """
        :type tickets: List[List[str]]
        :rtype: List[str]
        """
        if not tickets: return []
        graph = collections.defaultdict(list)
        for a, b in sorted(tickets)[::-1]:
            graph[a].append(b)

        dfs = ['JFK']
        res = []
        while dfs:
            cur_node = dfs[-1]
            if not graph[cur_node]:
                res.append(dfs.pop())
            else:
                dfs.append(graph[cur_node].pop())
        return res[::-1]