給定一個完美二叉樹,其所有葉子節點都在同一層,每個父節點都有兩個子節點。二叉樹定義如下:
struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每個 next 指針,讓這個指針指向其下一個右側節點。如果找不到下一個右側節點,則將 next 指針設置爲 NULL。
初始狀態下,所有 next 指針都被設置爲 NULL。
示例:
解題思路1:
這一題可以採用隊列層序遍歷的方法(指向下一個會有一個判斷,不能時當前層的最後一個,及1,3,7,他們都直接指向空)
- 1進隊列,1出隊列,1指向隊列下一個空,1左(2)進隊,1右(3)進隊
- 2出隊列,2指向隊列下一個3,2左(4)進隊,2右(5)進隊
- 3出隊列,3指向隊列下一個空,3左(6)進隊,3右(7)進隊
- 4出隊列,4指向隊列下一個5,3左(6)進隊,3右(7)進隊
- 隊列有4567,然後,逐個出隊列並指向下一個,最後7指向空,就完事了
提交代碼1:(隊列層序,Runtime: 64 ms, faster than 81.18 % )
class Solution:
def connect(self, root: 'Node') -> 'Node':
if not root:
return root
Q = collections.deque([root])
while Q:
size = len(Q)
for i in range(size):
node = Q.popleft()
if i < size - 1:
node.next = Q[0]
if node.left:
Q.append(node.left)
if node.right:
Q.append(node.right)
return root
解題思路2:
這一題不用隊列也能做,每次讓根節點的左節點指向右節點,如果根節點有下一個指向節點,那麼根節點右節點指向根節點下一個節點的左節點
提交代碼2:(Runtime: 64 ms, faster than 81.18 % )
class Solution:
def connect(self, root: 'Node') -> 'Node':
if not root:
return root
leftmost = root
while leftmost.left:
head = leftmost
while head:
head.left.next = head.right
if head.next:
head.right.next = head.next.left
head = head.next
leftmost = leftmost.left
return root