-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathstep-by-step-directions-from-a-binary-tree-node-to-another.py
44 lines (42 loc) · 1.56 KB
/
step-by-step-directions-from-a-binary-tree-node-to-another.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def getDirections(self, root: Optional[TreeNode], startValue: int, destValue: int) -> str:
def traverse(node):
if not node:
return None
lnode, rnode = None, None
if node.left:
parent[node.left] = node
lnode = traverse(node.left)
if node.right:
parent[node.right] = node
rnode = traverse(node.right)
if node.val == startValue:
return node
return lnode or rnode
parent = {}
start = traverse(root)
qu = deque()
qu.append([start, ''])
visited = set()
visited.add(start)
while qu:
l = len(qu)
for _ in range(l):
node, path = qu.popleft()
if node.val == destValue:
return path
if node.left and node.left not in visited:
qu.append((node.left, path+'L'))
visited.add(node.left)
if node.right and node.right not in visited:
qu.append((node.right, path+'R'))
visited.add(node.right)
if node in parent and parent[node] not in visited:
qu.append((parent[node], path+'U'))
visited.add(parent[node])