不会就坚持48天吧 展平二叉树

加油

中序遍历之后生成新的树
算法

题目要求我们返回按照中序遍历的结果改造而成的、只有右节点的等价二叉搜索树。我们可以进行如下操作:

先对输入的二叉搜索树执行中序遍历,将结果保存到一个列表中;

然后根据列表中的节点值,创建等价的只含有右节点的二叉搜索树,其过程等价于根据节点值创建一个链表。

class Solution {
    public TreeNode increasingBST(TreeNode root) {
        List<Integer> res = new ArrayList<Integer>();
        inorder(root, res);

        TreeNode dummyNode = new TreeNode(-1);
        TreeNode currNode = dummyNode;
        for (int value : res) {
            currNode.right = new TreeNode(value);
            currNode = currNode.right;
        }
        return dummyNode.right;
    }

    public void inorder(TreeNode node, List<Integer> res) {
        if (node == null) {
            return;
        }
        inorder(node.left, res);
        res.add(node.val);
        inorder(node.right, res);
    }
}

第二个

展平二叉搜索树,也就是对二叉搜索树进行中序遍历,然后将各个节点的指针进行修改,从而形成一棵只有右子树的树,解题思路如下:

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

class Solution {
    // cur 指向重排后的树的最后一个节点
    private TreeNode cur;
    public TreeNode increasingBST(TreeNode root) {
        // 创建一个辅助头节点,cur 指向辅助头节点
        TreeNode head = new TreeNode(0,null, null);
        cur = head;
        // 进行中序遍历 DFS
        dfs(root);
        // 返回树的第一个节点。
        return head.right;
    }
    private void dfs(TreeNode node){
        // 递归终止条件:node 为空
        if(node == null) return;

        // 中序遍历:左->中->右
        dfs(node.left);
        // 将当前节点插入重排的树的右子树中,同时更新重排树的最后一个节点。
        cur.right = node;
        cur = cur.right;
        // 将节点的左边节点设置为 null。
        node.left = null;
        dfs(node.right);
    }
}

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值