分治-二叉搜索树与双向链表-JZ26

描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。如下图所示
图片来自牛客网

注意:
1.要求不能创建任何新的结点,只能调整树中结点指针的指向。当转化完成以后,树中节点的左指针需要指向前驱,树中节点的右指针需要指向后继
2.返回链表中的第一个节点的指针
3.函数返回的TreeNode,有左右指针,其实可以看成一个双向链表的数据结构
4.你不用输出或者处理,示例中输出里面的英文,比如"From left to right are:"这样的,程序会根据你的返回值自动打印输出

示例:

输入: {10,6,14,4,8,12,16}
输出:From left to right are:4,6,8,10,12,14,16;From right to left are:16,14,12,10,8,6,4;
解析:
输入就是一棵二叉树,如上图,输出的时候会将这个双向链表从左到右输出,以及 从右到左输出,确保答案的正确

示例1

输入: {10,6,14,4,8,12,16}
返回值: From left to right are:4,6,8,10,12,14,16;From right to left are:16,14,12,10,8,6,4;

示例2

输入: {5,4,#,3,#,2,#,1}
返回值: From left to right are:1,2,3,4,5;From right to left are:5,4,3,2,1;
说明:树只有左子树
5
/
4
/
3
/
2
/
1 树的形状如上图

思路
1.使用二叉树的中序遍历可以获得一个要求链表结点的访问顺序;
2.在这个顺序中,我们要存储前一个节点的结点cur(双链表尾结点)和当前的结点pre(cur的值一定是比pre小的),
3.修改pre和cur的左右指向,cur.right = pre; pre.left = cur;
中序遍历核心代码

Node pre;
Node cur;
dfs(root){
   // 递归左节点
   dfs(root.left);
   // 处理结点指针
   cur.right = pre;
   pre.left = cur;
   cur = pre;
   // 递归右节点
   dfs(root.right);
}

代码
使用全局成员变量lastNodeInList 表示遍历的双连标的尾结点,参数pRootOfTree就表示当前结点,修改两个节点的结点的左右指针指向,注意防止NPE。

/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
	//双链表尾结点
    TreeNode lastNodeInList = null;
    public TreeNode Convert(TreeNode pRootOfTree) {
        convertNode(pRootOfTree);
        
        TreeNode headNodeInList = lastNodeInList;
        while (headNodeInList != null && headNodeInList.left != null) {
            headNodeInList = headNodeInList.left;
        }
        return headNodeInList;
    }
    
    private void convertNode(TreeNode pRootOfTree) {
        if (pRootOfTree == null) {
            return;
        }
        if (pRootOfTree.left != null) {
            convertNode(pRootOfTree.left);
        }
        
        pRootOfTree.left = lastNodeInList;
        if (lastNodeInList != null) {
            lastNodeInList.right = pRootOfTree;
        }
        lastNodeInList = pRootOfTree;
        
        if (pRootOfTree.right != null) {
            convertNode(pRootOfTree.right);
        }
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值