- 题目
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。 - 思路
二叉搜索树的左子树的根节点小于中间节点,右子树的根节点大于中间节点;因此,可以参考中序遍历,遍历二叉树。
1.设置两个变量head和realHead,head存储当前子树的根节点,readHead存储二叉树拥有最小节点值的节点,(二叉搜索树中序遍历的第一个节点)。
2.每次遍历到当前节点pRootOfTree时,将上次遍历的head的右指针指向pRootOfTree。pRootOfTree的左指针指向head,然后更新head,令head = pRootOfTree。
3.应用中序遍历,重复上述过程。
树结构
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
核心代码
//将二叉搜索树转换成排序的双向链表
public class Solution {
private TreeNode head = null;
private TreeNode realHead = null;
//将每一棵子树转换成链表
public void ConvertSub(TreeNode pRootOfTree){
if(pRootOfTree == null) return;
ConvertSub(pRootOfTree.left);
//设置初始状态
if(head == null){
head = pRootOfTree;
realHead = pRootOfTree;
}else{
//转换成双向链表
head.right = pRootOfTree;
pRootOfTree.left = head;
head = pRootOfTree;
}
ConvertSub(pRootOfTree.right);
}
public TreeNode Convert(TreeNode pRootOfTree) {
ConvertSub(pRootOfTree);
return realHead;
}
}