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

题目描述

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

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);
		
	}
}