[LeetCode]题解（python）：116-Populating Next Right Pointers in Each Node

题目来源：

　　https://leetcode.com/problems/populating-next-right-pointers-in-each-node/

---

 

题意分析：

　　给定一个二叉完全树，将同一层的节点连起来，其中最后一个节点连接到NULL。如

     　　1
       /  \
      2    3
     / \  / \
    4  5  6  7
得到：

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

---

 

题目思路：

　　由于是完全树。首先，将所有节点的左子树连接到右子树；然后将节点的右子树连接到下一个节点的左节点。

---

 

代码（python）：

　　

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

class Solution(object):
    def connect(self, root):
        """
        :type root: TreeLinkNode
        :rtype: nothing
        """
        if root == None or root.left == None:
            return
        root.left.next = root.right
        if root.next:
            root.right.next = root.next.left
        self.connect(root.left)
        self.connect(root.right)