【剑指offer】二叉搜索树与双向链表

题目描述

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

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

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

    }

}
*/
public class Solution {
    public TreeNode Convert(TreeNode pRootOfTree) {
        if(pRootOfTree==null)
            {
            return null;
        }
        //记录链表的最后一个节点
        TreeNode lastNodeInList = null;
        lastNodeInList=convertToList(pRootOfTree,lastNodeInList);
        
        //找到链表的头节点
        TreeNode listHead = lastNodeInList;
        while(listHead!=null&&listHead.left!=null)
            {
            listHead = listHead.left;
        }
        return listHead;
        
    }
    
    public TreeNode convertToList(TreeNode root,TreeNode lastNodeInList)
        {
       
        if(root==null)
            { 
            return null;
        }
        //中序遍历，所以先处理左子树，再处理当前节点，最后处理右子树
        //记录当前节点
        TreeNode current = root;
        //递归处理左子树
        if(current.left!=null)
            {
            lastNodeInList = convertToList(root.left,lastNodeInList);
        }
        //处理当前节点
        //当前节点的左指针指向已经转换成链表的部分的最后一个节点
        current.left = lastNodeInList;
        //已经转换好成链表的部分的最后一个节点的右指针指向当前节点
        if(lastNodeInList!=null)
            {
            lastNodeInList.right = current;
        }
        //更新链表的最后一个节点
        lastNodeInList = current;
        
        //递归处理右子树
        if(current.right!=null)
            {
            lastNodeInList = convertToList(root.right,lastNodeInList);
        }
        return lastNodeInList;
       
    }
}