[leetcode] 897. Increasing Order Search Tree

二叉排序树转换为仅有右结点的排序树

原创已于 2024-04-26 19:48:08 修改 · 186 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #职场和发展

于 2020-10-22 20:23:17 首次发布

leetcode题解同时被 2 个专栏收录

903 篇文章

订阅专栏

python

556 篇文章

订阅专栏

该博客介绍了如何将一个二叉排序树重新排列，使其成为每个节点只有右孩子的形式。提供了两种方法，一种是通过中序遍历重新构造，另一种是递归地使用一个辅助节点。代码示例展示了如何实现这一转换，最后返回新的根节点。

Description

Given a binary search tree, rearrange the tree in in-order so that the leftmost node in the tree is now the root of the tree, and every node has no left child and only 1 right child.

Example 1:

Example 1:
Input: [5,3,6,2,4,null,8,1,null,null,null,7,9]

       5
      / \
    3    6
   / \    \
  2   4    8
 /        / \ 
1        7   9

Output: [1,null,2,null,3,null,4,null,5,null,6,null,7,null,8,null,9]

 1
  \
   2
    \
     3
      \
       4
        \
         5
          \
           6
            \
             7
              \
               8
                \
                 9

Constraints:

The number of nodes in the given tree will be between 1 and 100.
Each node will have a unique integer value from 0 to 1000.

分析

题目的意思是：给定一个二叉排序树，然后变成一个只有右结点的二叉排序树。最简单的做法就是二叉树的中序遍历以后再构造一个只有右结点的二叉排序树就行了。还有另一种递归的解法，我比较欣赏，利用一个空结点cur来指向根结点，然后进行中序遍历，对于左子树，先递归，然后cur的右结点指向左子树，然后cur指向当前节点，再遍历右结点。比较抽象。最后返回该空结点的右节点就是结果了。

代码

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    pre = None

    def inorder(self,root):
        if not root:
            return 
        self.inorder(root.left)
        root.left=None
        self.pre.right = root
        self.pre = root
        self.inorder(root.right)


    def increasingBST(self, root: TreeNode) -> TreeNode:
        dummy = TreeNode(-1)
        dummy.right = root
        self.pre = dummy
        self.inorder(root)
        return dummy.right