Skip to content

Instantly share code, notes, and snippets.

@cocomoff
Created May 1, 2020 05:50
Show Gist options
  • Star 1 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save cocomoff/039feb9e320cd9aee88d6701fb745f73 to your computer and use it in GitHub Desktop.
Save cocomoff/039feb9e320cd9aee88d6701fb745f73 to your computer and use it in GitHub Desktop.
python bidirectional search
# Implementation https://stackoverflow.com/questions/54437905/bidirectional-search using deque
class Node:
def __init__(self, val, neighbors=[]):
self.val = val
self.neighbors = neighbors
self.visited_right = False # whether the node was reached by the BFS that started from source
self.visited_left = False # whether the node was reached by the BFS that started from destination
self.parent_right = None # used for retrieving the final path from source to the meeting point
self.parent_left = None # used for retrieving the final path from the meeting point to destination
# class Queue:
# pass # implement it yourself
from collections import deque
def bidirectional_search(s, t):
def extract_path(node):
"""return the path when both BFS's have met"""
node_copy = node
path = []
while node:
path.append(node.val)
node = node.parent_right
path.reverse()
del path[-1] # because the meeting node appears twice
while node_copy:
path.append(node_copy.val)
node_copy = node_copy.parent_left
return path
q = deque([])
q.append(s)
q.append(t)
s.visited_right = True
t.visited_left = True
while len(q) > 0:
n = q.pop()
if n.visited_left and n.visited_right: # if the node visited by both BFS's
return extract_path(n)
for node in n.neighbors:
if n.visited_left == True and not node.visited_left:
node.parent_left = n
node.visited_left = True
q.append(node)
if n.visited_right == True and not node.visited_right:
node.parent_right = n
node.visited_right = True
q.append(node)
# not found
return False
n0 = Node(0)
n1 = Node(1)
n2 = Node(2)
n3 = Node(3)
n4 = Node(4)
n5 = Node(5)
n6 = Node(6)
n7 = Node(7)
n0.neighbors = [n1, n5]
n1.neighbors = [n0, n2, n6]
n2.neighbors = [n1]
n3.neighbors = [n4, n6]
n4.neighbors = [n3]
n5.neighbors = [n0, n6]
n6.neighbors = [n1, n3, n5, n7]
n7.neighbors = [n6]
print(bidirectional_search(n0, n4))
@bemnet16
Copy link

bemnet16 commented Apr 6, 2023

Thanks!!!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment