一、数组 VS 二叉树
1.有序数组转换为二叉搜索树
思路:递归 时间复杂度O(N) 空间复杂度O(1)
nums为空,return None nums非空,nums[n/2]为中间元素,根结点,nums[:mid]为左子树, nums[mid+1:]为右子树
class Solution: def sortedArrayToBST(self, nums): if len(nums) == 0: return None mid = len(nums) // 2 root = TreeNode(nums[mid]) root.left = self.sortedArrayToBST(nums[0:mid]) root.right = self.sortedArrayToBST(nums[mid+1:]) return root
二、链表 VS 二叉树
1.有序链表转换为二叉搜索树 leetcode109
思路1:利用快慢指针的思想寻找链表中点,将其分为两个部分,再继续找左右两个新链表的中点。 当慢指针不与出发点重合时,说明这个子链的长度<3,此时只要把子链的右边节点(如果有的话) 作为左节点的右子树即可 时间复杂度O(N) 空间复杂度O(1)
class Solution: def sortedListToBST(self, head): def getMid(node): if not node: return slow = node fast = node pre_slow = None while fast.next and fast.next.next: fast = fast.next.next pre_slow = slow slow = slow.next if slow == node:#中心点=head,没有左子树 return None,slow #返回左,中心点 else: #中心点!=None pre_slow.next = None#左子树到slow前一个节点结束 return node,slow left_head,mid_node = getMid(head) root = TreeNode(mid_node.val) #中心点 right_head = mid_node.next root.left = self.sortedListToBST(left_head) root.right = self.sortedListToBST(right_head) return root
2.二叉树转换为单向链表
题目:给定一个二叉树,将该二叉树就地转换为单链表,单链表中节点顺序为二叉树前序遍历顺序。
思路1:前序遍历二叉树,将节点指针push进入vector, 顺序遍历vector中的节点,链接相邻两节点,形成单链表。(不满足就地转换的条件)
思路2:
3.二叉树转换为双向链表
题目:输入一棵二元查找树,将该二元查找树转换成一个排序的双向链表。
要求不能创建任何新的节点,只能调整节点的指向
思路:中序遍历假设当前遍历的节点为root,root的左子树已经被转换为双向链表,使用两个变量pHead与pEnd分别指向链表的头结点与尾结点。那么在遍历root节点的时候,只需要将root节点的lchild指向pEnd,把pEnd的rchild指向root,此时root节点就被加入到双向链表里了,因此,root变成了双向链表的尾结点。
class TreeNode: def __init__(self, x): self.val = x self.left = None self.right = None class Test: def __init__(self): self.pHead = None self.pEnd = None def inOrderBSTree(self,root): if root == None: return self.inOrderBSTree(root.left) root.left = self.pEnd #使当前节点的左孩子指向双向链表中最后一个节点 if self.pEnd == None:#双向列表为空,当前遍历的节点为双向链表的头结点 self.pHead = root else:#使双向链表中最后一个节点的右孩子指向当前节点 self.pEnd.rchild = root self.pEnd = root#将当前节点设为双向链表中最后一个节点 self.inOrderBSTree(root.right)