leetcode 332. 重新安排行程——回溯

332. 重新安排行程

给你一份航线列表 tickets ，其中 tickets[i] = [$from_i$, $to_i$] 表示飞机出发和降落的机场地点。请你对该行程进行重新规划排序。

所有这些机票都属于一个从 JFK（肯尼迪国际机场）出发的先生，所以该行程必须从 JFK 开始。如果存在多种有效的行程，请你按字典排序返回最小的行程组合。

• 例如，行程 ["JFK", "LGA"] 与 ["JFK", "LGB"] 相比就更小，排序更靠前。

假定所有机票至少存在一种合理的行程。且所有的机票 必须都用一次 且 只能用一次。

示例 1：

输入：tickets = [["MUC","LHR"],["JFK","MUC"],["SFO","SJC"],["LHR","SFO"]]

输出：["JFK","MUC","LHR","SFO","SJC"]

示例 2：

输入：tickets = [["JFK","SFO"],["JFK","ATL"],["SFO","ATL"],["ATL","JFK"],["ATL","SFO"]]

输出：["JFK","ATL","JFK","SFO","ATL","SFO"]

解释：另一种有效的行程是 ["JFK","SFO","ATL","JFK","ATL","SFO"] ，但是它字典排序更大更靠后。

题目链接：332. 重新安排行程 - 力扣（LeetCode） (leetcode-cn.com)

解题思路

回溯

• 首先由题意可知，行程一定是从"JFK"出发
• 从当前位置出发遍历能到达的目的地，一次次递归找到最后的叶子节点（即最终的目的地）
• 怎么保存当前位置可以到达的目的地
  • 使用字典来保存
  • 遍历所有机票信息

class Solution:
    def findItinerary(self, tickets: List[List[str]]) -> List[str]:
        # 得到机票的数量
        n = len(tickets)
        # 创建一个字典来存放所有的起始点的目的地
        todest_dict = defaultdict(list)
        # 扫描所有的机票
        for ticket in tickets:
            todest_dict[ticket[0]].append(ticket[1])
        # 初始化path数列，用于返回最后的行程
        path = ["JFK"]
        
        def backtrack(path):
            # 终止条件
            if len(path) == n+1:
                return True
            # 还未完成完整路径
            # 先对数组排序，方便在有多种安排的情况下按字典序返回最小的组合
            todest_dict[path[-1]].sort()
            # 单层遍历
            for point in todest_dict[path[-1]]:
                # 处理节点
                # 得到并删除该目的地
                endpoint = todest_dict[path[-1]].pop(0)
                path.append(endpoint)
                # 递归
                if backtrack(path):
                    return True
                path.pop()
                todest_dict[path[-1]].append(endpoint)
        
        backtrack(path)
        return path

标签：JFK,tickets,ATL,行程,332,SFO,重新安排,path,leetcode
来源： https://www.cnblogs.com/waitting975/p/15380628.html