【LeetCode】116. Populating Next Right Pointers in Each Node 解題報告（Python）

給定一個完美二叉樹，其所有葉子節點都在同一層，每個父節點都有兩個子節點。二叉樹定義如下：

struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每個 next 指針，讓這個指針指向其下一個右側節點。如果找不到下一個右側節點，則將 next 指針設置爲 NULL。

初始狀態下，所有 next 指針都被設置爲 NULL。

示例：

解題思路1：

這一題可以採用隊列層序遍歷的方法（指向下一個會有一個判斷，不能時當前層的最後一個，及1，3，7，他們都直接指向空）

1. 1進隊列，1出隊列，1指向隊列下一個空，1左（2）進隊，1右（3）進隊
2. 2出隊列，2指向隊列下一個3，2左（4）進隊，2右（5）進隊
3. 3出隊列，3指向隊列下一個空，3左（6）進隊，3右（7）進隊
4. 4出隊列，4指向隊列下一個5，3左（6）進隊，3右（7）進隊
5. 隊列有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