剑指Offer-二叉搜索树与双向链表
题目描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
思路
思路一:
- 由于要求链表是有序的,可以借助二叉树中序遍历,因为中序遍历算法的特点就是从小到大访问结点。中序遍历过程中,根节点不断加到右边,这样可以保持从左到右升序。
- 由于中序遍历过程正好是转换成链表的过程,即可采用递归处理。
代码实现
package Tree; /** * 二叉搜索树与双向链表 * 输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。 */ public class Solution38 { //双向链表的左边头结点和右边头节点 TreeNode leftHead = null; TreeNode rightHead = null; /** * 中序遍历,递归 * * @param pRootOfTree * @return */ public TreeNode Convert(TreeNode pRootOfTree) { //递归调用叶子节点的左右节点返回null if (pRootOfTree == null) return null; //第一次运行时,它会使最左边叶子节点为链表第一个节点 Convert(pRootOfTree.left); if (rightHead == null) { leftHead = rightHead = pRootOfTree; } else { //把根节点插入到双向链表右边,rightHead向后移动 rightHead.right = pRootOfTree; pRootOfTree.left = rightHead; rightHead = pRootOfTree; } //把右叶子节点也插入到双向链表(rightHead已确定,直接插入) Convert(pRootOfTree.right); //返回左边头结点 return leftHead; } public class TreeNode { int val = 0; TreeNode left = null; TreeNode right = null; public TreeNode(int val) { this.val = val; } } }