题目描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
题目分析
二叉搜索树,中序遍历为递增的有序序列。
树的结点结构:lchild data rchild
双链表结点结构:pre data next
转换为双链表,原左孩子指针指向当前节点前驱,右孩子指向当前节点后继
二叉树中序遍历有两种方式,递归和非递归
非递归
import java.util.*;
public class Solution {
public TreeNode Convert(TreeNode pRootOfTree) {
Stack<TreeNode> stack = new Stack<TreeNode>();
//空树,返回空
if(pRootOfTree==null)
return null;
TreeNode p = pRootOfTree; //当前节点
TreeNode pre = null;//当前节点的前驱
TreeNode head = null;
boolean flag = true;
//1.栈不空或者p不空,循环
while(!stack.empty()||p!=null){
//一路向左,入栈
while(p!=null){
stack.push(p); //入栈
p = p.left;
}
//当前节点出栈
p = stack.pop();
//第一次出栈点,即双链表第一个节点无前驱,返回头节点
if(flag){
head = p;
pre = p;
flag = false;
}
//除第一个点外
else{
pre.right = p; //pre后继是p
p.left = pre;
pre = p;
}
p = p.right;
}
return head;
}
}
递归
//中序遍历
//二叉树,左孩子前驱,右孩子后继,类似于线索二叉树
//中序遍历递归/非递归
//递归写法
public class Solution {
public TreeNode Convert(TreeNode pRootOfTree) {
if(pRootOfTree==null)
return null;
//若为叶子结点,返回
if(pRootOfTree.left==null && pRootOfTree.right==null)
return pRootOfTree;
//1.返回双链表的头节点——从根结点开始一路向左,树最左边的结点
TreeNode left= Convert(pRootOfTree.left);
TreeNode cur = left;//当前双链表第一个结点
//2.左子树双链表最后一个结点,根节点左孩子一路向右
while(cur!=null&&cur.right!=null){
cur = cur.right;
}
//3.左子树链表不空,当前root放入左子树链表中
if(left!=null){
cur.right = pRootOfTree;
pRootOfTree.left = cur;
}
//4.右子树构造为双链表并返回头节点
TreeNode right = Convert(pRootOfTree.right);
TreeNode cur2 = right;
//5.右子树链表不空,当前节点放入右子树链表中
if(right!=null){
cur2.left = pRootOfTree;
pRootOfTree.right = cur2;
}
if(left!=null)
return left;
else
return pRootOfTree;
}
}