二叉搜索树转换成双向链表——剑指offer

来源:互联网 发布:冒泡排序c语言详解 编辑:程序博客网 时间:2024/06/07 02:46

题目描述

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

public class TreeToList {public class TreeNode {    int val = 0;    TreeNode left = null;    TreeNode right = null;    public TreeNode(int val) {        this.val = val;    }}public TreeNode Convert(TreeNode pRootOfTree) {TreeNode lastNode=null;lastNode = ConvertNode(pRootOfTree, lastNode);// 注意这里的lastNode作为参数时并没有分配内存,所以不是一个引用,所以要用返回值获取它的值TreeNode headOfList=lastNode;while(headOfList!=null && headOfList.left!=null)//返回值是链表尾,所以找到链表头返回headOfList=headOfList.left;return headOfList;}private TreeNode ConvertNode(TreeNode pRootOfTree, TreeNode lastNode) {if(pRootOfTree == null)return null;TreeNode currentNode=pRootOfTree;//当前节点为根节点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;//返回链表的尾节点}public static void main(String[] args){TreeToList t=new TreeToList();TreeNode root=t.new TreeNode(10);TreeNode r6=t.new TreeNode(6);TreeNode r4=t.new TreeNode(4);TreeNode r8=t.new TreeNode(8);TreeNode r14=t.new TreeNode(14);TreeNode r12=t.new TreeNode(12);TreeNode r16=t.new TreeNode(16);root.left=r6;root.right=r14;r6.left=r4;r6.right=r8;r14.left=r12;r14.right=r16;TreeNode res=t.Convert(root);}}


0 0
原创粉丝点击