Advertisement
mhr007

next node none

Oct 22nd, 2021
42
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.17 KB | None | 0 0
  1. """
  2. # Definition for a Node.
  3. class Node:
  4. def __init__(self, val: int = 0, left: 'Node' = None, right: 'Node' = None, next: 'Node' = None):
  5. self.val = val
  6. self.left = left
  7. self.right = right
  8. self.next = next
  9. """
  10.  
  11. class Solution:
  12.  
  13.  
  14. def travarse (self, node,q):
  15.  
  16. if node == None:
  17. return None
  18.  
  19.  
  20.  
  21.  
  22.  
  23. def connect(self, root: 'Node') -> 'Node':
  24.  
  25.  
  26. q = []
  27. q.append(root)
  28. q1 = []
  29. while len(q)>0:
  30.  
  31. node = q.pop(0)
  32. if node.left != None:
  33. q1.append(node.left)
  34.  
  35.  
  36. if node.right != None:
  37. q1.append(node.right)
  38.  
  39.  
  40. if len(q)==0:
  41. node = q1.pop(0)
  42.  
  43. while len(q1)>0:
  44. node2 = q1.pop(0)
  45.  
  46. node.next = node2
  47. node = node2
  48. q.append(node2)
  49.  
  50.  
  51.  
  52.  
  53.  
  54.  
  55.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement