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

题目来源：

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

题意分析：

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

         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)

巴特西

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

最新文章

热门文章