LeetCode 116. 填充同一層的兄弟節點（Populating Next Right Pointers in Each Node）

• 原題
• 思路
• 程式碼

原題

給定一個二叉樹

struct TreeLinkNode {
TreeLinkNode *left;
TreeLinkNode *right;
TreeLinkNode *next;
}
填充它的每個 next 指標，讓這個指標指向其下一個右側節點。如果找不到下一個右側節點，則將 next 指標設定為 NULL。

初始狀態下，所有 next 指標都被設定為 NULL。

說明:

你只能使用額外常數空間。
使用遞迴解題也符合要求，本題中遞迴程式佔用的棧空間不算做額外的空間複雜度。
你可以假設它是一個完美二叉樹（即所有葉子節點都在同一層，每個父節點都有兩個子節點）。
示例:

給定完美二叉樹，

 	1
   / \
  2   3
 / \  / \
4  5  6  7

呼叫你的函式後，該完美二叉樹變為：

 	1 -> NULL
   / \
 2 -> 3 -> NULL
 / \  / \
4->5->6->7 -> NULL

思路

由於題目中常數空間的限制，採用佇列的方式對樹進行層次遍歷是不可行的。
若把問題轉化為給定一層已經連結的節點，請連結下面一層，這樣問題就迎刃而解了。用start記錄當前層的第一個節點，再用cur迴圈當前層的每個節點。若cur.left存在，則連結cur.left和cur.right；若cur.next存在，則連結cur.right和cur.next.left。下一層的首個節點即start.left，如此迴圈。
按此思路，程式碼也就很簡單了。

程式碼

# Definition for binary tree with next pointer.
# class TreeLinkNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
#         self.next = None

class Solution:
    # @param root, a tree link node
    # @return nothing
    def connect
 
(self, root):
        start = root
        while start:
            cur = start
            while cur and cur.left:
                cur.left.next = cur.right
                if cur.next:
                    cur.right.next = cur.next.left
                cur = cur.next
            start = start.left