BFS之公交路线

leetcode 815. 公交路线

我们有一系列公交路线。每一条路线 routes[i] 上都有一辆公交车在上面循环行驶。例如，有一条路线 routes[0] = [1, 5, 7]，表示第一辆 (下标为0) 公交车会一直按照 1->5->7->1->5->7->1->… 的车站路线行驶。

假设我们从 S 车站开始（初始时不在公交车上），要去往 T 站。 期间仅可乘坐公交车，求出最少乘坐的公交车数量。返回 -1 表示不可能到达终点车站。

示例：

输入：
routes = [[1, 2, 7], [3, 6, 7]]
S = 1
T = 6
输出：2
解释：
最优策略是先乘坐第一辆公交车到达车站 7, 然后换乘第二辆公交车到车站 6。

提示：

1 <= routes.length <= 500.
1 <= routes[i].length <= 10^5.
0 <= routes[i][j] < 10 ^ 6.

思路
广度优先搜索

• 首先根据公交路线进行构图，构图原则为：如果路线i和路线j存在公共站点，则将二者之间连一条边，同时把所有和起点S存在交集的路线和S相连，也即此处是车站和路线进行连接，同理，把所有和终点T存在交集的路线和T相连．
• 然后进行广度优先搜索，第一层节点是与起点S相连的路线，初始化乘坐数量为1，并且将第一层标记为已访问，如果终点在第一层中的任何一条路线中，直接返回乘坐数量；否则继续遍历第二层节点（第一层节点的邻居节点），乘坐数量加1，依次类推，直到遍历完毕.
• 如果仍然没有到达终点，则返回-1

看代码

class Solution:
    def numBusesToDestination(self, routes: List[List[int]], S: int, T: int) -> int:
        n = len(routes)
        if S == T:
            return 0
            
        routes = list(map(set, routes))　# 转为set，加快查找速度
        
        graph = collections.defaultdict(set)
		# 构图，任何两个路线存在相同车站，就将二者相连
        for i, r1 in enumerate(routes):
            for j in range(i+1, n):
                r2 = routes[j]
                for r in r1:
                    if r in r2:
                        graph[i].add(j)
                        graph[j].add(i)
		
		# 将起点和终点与和其相连的路线进行连接
        visited, target = set(), set()
        for node, route in enumerate(routes):
            if S in route:
                visited.add(node)
            if T in route:
                target.add(node)
		
		# 第一层节点，初始化乘坐数量为1
        queue = [(node, 1) for node in visited]
        while queue:
            size = len(queue)
            for i in range(size):
                node, res = queue.pop(0)
                if node in target:　# 找到终点，返回
                    return res
                for neibor in graph[node]:　# 未到终点，遍历下一层
                    if neibor not in visited:
                        queue.append((neibor, res + 1))
                        visited.add(neibor)
        return -1　# 未找到终点