一刷239-剑指 Offer II 052. 展平二叉搜索树(同:897. 递增顺序搜索树)

题目:
给你一棵二叉搜索树,请 按中序遍历 将其重新排列为一棵递增顺序搜索树,
使树中最左边的节点成为树的根节点,并且每个节点没有左子节点,只有一个右子节点。

在这里插入图片描述

输入:root = [5,3,6,2,4,null,8,1,null,null,null,7,9]
输出:[1,null,2,null,3,null,4,null,5,null,6,null,7,null,8,null,9]

在这里插入图片描述

输入:root = [5,1,7]
输出:[1,null,5,null,7]
 
提示:
树中节点数的取值范围是 [1, 100]
0 <= Node.val <= 1000
--------------------
思路:
展平二叉搜索树,也就是对二叉搜索树进行中序遍历,然后将各个节点的指针进行修改,
从而形成一棵只有右子树的树,解题思路如下:

设置一个辅助头节点 head : 指向重排后的树。
对树进行中序遍历,将遍历到的节点插入重排后的树head的最后一个节点的右子树,
同时将当前遍历到的节点的左子树设置为 null。

遍历完后,直接返回 head.right 就是重排后的树的根节点。

注意:由于插入操作需要知道重排后的树 head 的最后一个节点,
所以可以设置一个节点 cur 指向重排后的树的最后一个节点。
-----------------------
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    private TreeNode cur;//cur 指向重排后的树的最后一个节点
    public TreeNode increasingBST(TreeNode root) {
        TreeNode head = new TreeNode(0, null, null);//创建一个辅助头节点,cur指向辅助头节点
        cur = head;//初始化
        dfs(root);//DFS 中序遍历
        return head.right;//返回树的第一个节点
    }
    private void dfs(TreeNode node) {
        if (node == null) return;//递归终止条件: node为空
        dfs(node.left);//中序遍历: 左--中--右
        cur.right = node;//将当前节点插入到重排后树的右子树,同时更新cur这个重排树的最后一个节点
        cur = cur.right;
        node.left = null;//将节点的左边节点设置为null
        dfs(node.right);
    }
}

LC

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值