剑指Offer-题36(Java版):二叉搜索树与双向链表

参考自:《剑指Offer——名企面试官精讲典型编程题》

题目:二叉搜索树与双向链表
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。

主要思路:在二叉搜索树中,左子节点的值小于父节点的值,右子节点的值大于父节点的值。因此,将二叉搜索树转换成一个排序双向链表时,左子节点相当于双向链表中的前一个节点,右子节点相当于双向链表中的后一个节点。
使用中序遍历,
首先转换左子树,并获取其最后一个节点(即左子树中的最大节点);然后,将该最大节点和父节点转换成链表;
最后,转换右子树,并将父节点与右子树的最小节点连接。
递归上述转换过程。其中,父节点每次都是分别与左子树中最大节点和右子树中最小节点进行连接

关键点:递归,有序,二叉搜索树与排序双向链表的对应关系

时间复杂度:O(n)

public class TreeToLinkedList {
    public static void main(String[] args) {
        TreeNode root = TreeNode.generateBinaryTree();
        TreeNode result = convert(root);
        printResult(result);
    }

    private static TreeNode convert(TreeNode root) {
        if (root == null) {
            return null;
        }
        //因为lastNode是尾节点,所以要向前遍历找到头结点
        TreeNode result = convertNode(root, null);
        while (result != null && result.left != null) {
            result = result.left;
        }
        return result;
    }

    private static TreeNode convertNode(TreeNode currentNode, TreeNode lastNode) {
        if (currentNode == null) {
            return null;
        }
        //转换左子树,并获取其最后一个节点(即左子树中的最大节点)
        if (currentNode.left != null) {
            lastNode = convertNode(currentNode.left, lastNode);
        }
        //左节点相当于链表中的前驱
        currentNode.left = lastNode;
        if (lastNode != null) {
            //右节点相当于链表中的后继
            lastNode.right = currentNode;
        }
        //更新尾节点
        lastNode = currentNode;

        //转换右子树,并将父节点与右子树的最小节点链接
        if (currentNode.right != null) {
            lastNode = convertNode(currentNode.right, lastNode);
        }
        return lastNode;
    }

    private static void printResult(TreeNode result) {
        TreeNode curNode = result;
        while (curNode != null) {
            System.out.println(curNode.val);
            //右节点就是链表的下一个节点
            curNode = curNode.right;
        }
    }
}
class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;

    TreeNode(int x) {
        val = x;
    }

    //            10
    //         /      \
    //        6        14
    //       /\        /\
    //      4  8     12  16

    /**
     * 生成二叉搜索树
     *
     * @return
     */
    public static TreeNode generateBinaryTree() {
        TreeNode root = new TreeNode(10);
        TreeNode node6 = new TreeNode(6);
        TreeNode node14 = new TreeNode(14);
        TreeNode node4 = new TreeNode(4);
        TreeNode node8 = new TreeNode(8);
        TreeNode node12 = new TreeNode(12);
        TreeNode node16 = new TreeNode(16);
        connectNode(root, node6, node14);
        connectNode(node6, node4, node8);
        connectNode(node14, node12, node16);
        return root;
    }

    private static void connectNode(TreeNode root, TreeNode left, TreeNode right) {
        root.left = left;
        root.right = right;
    }
}

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值