DEV Community

Abhishek Chaudhary
Abhishek Chaudhary

Posted on

1 1

Shortest Path Visiting All Nodes

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 contain i.
  • If graph[a] contains b, then graph[b] contains a.
  • 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
Enter fullscreen mode Exit fullscreen mode

AWS GenAI LIVE image

Real challenges. Real solutions. Real talk.

From technical discussions to philosophical debates, AWS and AWS Partners examine the impact and evolution of gen AI.

Learn more

Top comments (0)

Sentry image

See why 4M developers consider Sentry, “not bad.”

Fixing code doesn’t have to be the worst part of your day. Learn how Sentry can help.

Learn more

👋 Kindness is contagious

Please leave a ❤️ or a friendly comment on this post if you found it helpful!

Okay