题目:
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
思路:
将二叉搜索树转换成一个排序的双链表,利用二叉搜索树的特性,非空二叉树的左子树上的节点的值都比根节点的值小,右子树上的节点的值都比根节点的值大,利用搜索二叉树的中序遍历得到的就是一个排序的序列;
双链表不仅要指向它后面的节点还要指向它前面的节点,所以需要一个引用变量preNode,每次访问过的节点要保存在preNode中,这样才能得到一个节点前一个节点,二叉搜索树中节点的属性有left right val;排序双向链表的前一个节点对应left 后一个节点对应right,无新节点生成。
public TreeNode Convert(TreeNode pRootOfTree)
{
if (pRootOfTree==null)
{
return null;
}
LinkedList<TreeNode> stack = new LinkedList<TreeNode>();
TreeNode current = pRootOfTree;
TreeNode preNode = null;//用于保存前一个节点
boolean isFrist = true;//判断是不是第一个访问树,如果是的话,让当前节点作为双向链表的头节点
while(current != null || !stack.isEmpty())
{
while(current != null)
{
stack.push(current);
current = current.left;
}
//栈不为空 出栈栈顶节点并判断是否是第一次出栈元素
if(!stack.isEmpty())
{
current = stack.pop();
//第一次出栈节点,把该节点作为双向链表的头节点并将isFrist置为false
if(isFrist)
{
pRootOfTree = current;
preNode = pRootOfTree;
isFrist = false;
}
//若不是第一次出栈节点,前一个节点的right是当前节点 当前节点的left是前一个节点 访问完当前节点把它置为前一个节点
else
{
preNode.right = current;
current.left = preNode;
preNode = current;
}
current = current.right; //访问当前节点的右子树
}
}
return pRootOfTree;
}