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