剑指笔记——36.二叉搜索树与双向链表

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

思路:在转换的时候设置节点的左子树指向链表的前一个指针,右子树指向链表的后一个指针。我们将整个二叉树分为三个部分,根节点,左子树和右子树。将根节点和左子树中最大的元素链接,并且与右子树中最小的元素链接。然后才用递归的方式分别处理左子树和右子树。整个思路好复杂。。。。

代码:

public TreeNode Convert(TreeNode pRootOfTree) {
TreeNode lastlist = covertNode(pRootOfTree,null);//这部分代码用来返回双向链表的一个尾节点
TreeNode pHead = lastlist;
while (pHead != null && pHead.left != null) { //从最右的尾节点一直找到最左的头结点,然后返回头结点
pHead = pHead.left;
}
return pHead;
}
public TreeNode covertNode(TreeNode root, TreeNode lastlist) {
if (root == null)
return null;
TreeNode cur = root;
if (cur.left != null) {  //当前节点的左节点不为空时,递归调用可以找到左子树中最大的一个元素,返回该元素
lastlist = covertNode(cur.left,lastlist);
}
cur.left = lastlist;  //将根节点的左节点设置为该值
if (lastlist != null) {  //如果返回的这个节点不为空,将该节点的有节点设置为根节点
lastlist.right = cur;
}
lastlist = cur;
if (cur.right != null) {  //如果有节点不为空,递归调用返回右子树中最小的节点
lastlist = covertNode(cur.right,lastlist);
}
return lastlist;  //返回该子树形成的双向链表中最大的一个元素
}

猜你喜欢

转载自blog.csdn.net/chenxy132/article/details/89205845