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

题目来源：

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

---

题意分析：

　　根据上一题，如果给定的树不是完全树同层的连接。要求常数空间时间复杂度。如：

         1
       /  
      2    3
     /     
    4   5    7
结果是：

         1 -> NULL
       /  
      2 -> 3 -> NULL
     /     
    4-> 5 -> 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:
            tmp,tmp1,tmp2 = root,None,None
            while tmp:
                if tmp.left:
                    if tmp1:
                        tmp1.next = tmp.left
                    tmp1 = tmp.left
                    if not tmp2:
                        tmp2 = tmp1
                if tmp.right:
                    if tmp1:
                        tmp1.next = tmp.right
                    tmp1 = tmp.right
                    if not tmp2:
                        tmp2 = tmp1
                tmp = tmp.next
            self.connect(tmp2)