【剑指Offer-Java】二叉搜索树与双向链表

题目描述

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

思路

关于树的问题一般都可以用递归解决

因为是二叉搜索树，所以其中序遍历就是最终想要的排序链表的元素顺序
又因为要实现双向，所以要定义两个表头：左表头的和右表头，最终返回左表头即可

递归思想：先遍历左子树，找到左子树最两边的结点，即为左边子链表的首尾结点（遍历右子树同理）
继续遍历剩下的左子树，找到更内圈的首尾结点…直到最终只剩一个子树的根结点了

中序遍历过程：
（1）先递归遍历左子树
（2）递归的具体过程：
递归出口：遍历到最后，某子树中只剩一个根结点，此时根结点就是左右表头
把根结点插入到双向链表右边，右表头向后移动,从而得到升序链表
（3）递归遍历右子树

实现

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

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

    }

}
*/
public class Solution {
        //定义双向链表的左右表头(多个引用指向一个对象，不是创建新结点)
        //双向链表的首尾结点（递归过程中是子链表的首尾结点）
        TreeNode leftHead=null;    
        TreeNode rightHead=null;
    public TreeNode Convert(TreeNode pRootOfTree) {
        //特殊情况:空树
        if(pRootOfTree==null) return null;
        
        //递归遍历左子树
        Convert(pRootOfTree.left);
        
        //遍历根结点
        if(rightHead==null)    {
            leftHead=rightHead=pRootOfTree;
        }
        else{   //把根结点插入到双向链表右边，右表头向后移动,从而得到升序链表
            //前两句实现双向！！
            rightHead.right=pRootOfTree;
            pRootOfTree.left=rightHead;  
            //跳到下一个结点
            rightHead=pRootOfTree;
        }
        
        //递归遍历右子树,把右叶子节点也插入到双向链表（rightHead已确定，直接插入）
        Convert(pRootOfTree.right);
        
        //返回双向链表的左表头
        return leftHead;
        
    }
}

Notes:左右表头的定义要放在递归函数外面