二叉搜索树转换为双向链表

https://blog.nowcoder.net/n/4fa351e14ee64514babb6742ee023627

 

题意整理

• 输入一棵二叉搜索树，将该二叉搜索树转换成一个排序的双向链表。

方法一（递归）

1.解题思路

由于二叉搜索树的中序遍历是从小到大依次输出的，所以可以利用中序遍历，在遍历的过程中，逐个改变当前节点的指向。

• 预先定义一个pre指针，指向当前节点的前一个节点。以及一个head指针指向头节点。
• 当pre为空时，可以确定当前节点为双向链表的头节点head。其它情况下则可以将pre的后继指向当前节点cur。
• 遍历过程中，对于每一个cur，都应将其前驱指向pre。同时不断跟新pre。最后返回head即可。

图解展示： 

2.代码实现

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

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

    }

}
*/
public class Solution {
    //pre记录当前节点前一个，head记录头节点
    TreeNode pre=null,head=null;
    
    public TreeNode Convert(TreeNode pRootOfTree) {
        //为空判断
        if(pRootOfTree==null) return null;
        //递归
        dfs(pRootOfTree);
        return head;
        
    }
    
    private void dfs(TreeNode cur){
        //递归终止条件
        if(cur==null) return;
        //遍历左子树
        dfs(cur.left);
        //如果pre为空，说明当前是头节点
        if(pre==null) head=cur;
        //将pre的right指针指向cur
        else pre.right=cur;
        //将cur的left指针指向pre
        cur.left=pre;
        //更新pre
        pre=cur;
        //遍历右子树
        dfs(cur.right);
    }
}