You have an undirected, connected graph of n nodes labeled from 0 to n - 1. You are given an array graph where graph[i] is a list of all the nodes connected with node i by an edge.
Return the length of the shortest path that visits every node. You may start and stop at any node, you may revisit nodes multiple times, and you may reuse edges.
Example 1:
Input: graph = [[1,2,3],[0],[0],[0]]
Output: 4
Explanation: One possible path is [1,0,2,0,3]
Example 2:
Input: graph = [[1],[0,2,4],[1,3,4],[2],[1,2]]
Output: 4
Explanation: One possible path is [0,1,4,2,3]
Constraints:
-
n == graph.length -
1 <= n <= 12 -
0 <= graph[i].length <Â n -
graph[i]does not containi. - If
graph[a]containsb, thengraph[b]containsa. - The input graph is always connected.
SOLUTION:
class Solution:
def shortestPathLength(self, graph: List[List[int]]) -> int:
n = len(graph)
if n == 1:
return 0
req = (1 << n) - 1
visited = {}
queue = []
for i in range(n):
queue.append((0 | (1 << i), i))
lvl = 1
qi = 0
while qi < len(queue):
k = len(queue) - qi
while k > 0:
curr, currnode = queue[qi]
qi += 1
for u in graph[currnode]:
newMask = curr | (1 << u)
if newMask == req:
return lvl
if newMask in visited.get(u, {}):
continue
visited[u] = visited.get(u, set()).union({newMask})
queue.append((newMask, u))
k -= 1
lvl += 1
return -1
Top comments (0)