题目:输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
思路:先想一想最简单的情况怎么处理。即三个节点,根,左,右。这个时候应该先把根的left节点的right属性指向根,然后把根的right节点的left属性指向根,然后返回根的left节点。若根的left节点不是叶子节点,那么则以这个节点开始,求其叶子节点,并把根节点的left指向其尾,其尾的right指向根。说起来很抽象,直接看代码:
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
public TreeNode Convert(TreeNode pRootOfTree) {
if(pRootOfTree!=null){
return getHead(pRootOfTree);
}
return null;
}
public TreeNode getHead(TreeNode tree){
TreeNode head = tree;//假设双链表的头结点是tree,只有在tree没有左子树时成立。
if(tree.left!=null){//如果有左子树
if(tree.left.left==null&&tree.left.right==null){//如果这个节点是最小叶子节点,那么就是我们讨论的最简情况
head = tree.left;
head.right = tree;
}else{//否则
head = getHead(tree.left);//找到这个左子树的头结点
TreeNode temp = head;
while(temp.right!=null){
temp = temp.right;
}
temp.right = tree;
tree.left = temp;//并且找到这个左子树形成双向链表的尾节点,然后和root相关联
}
}
if(tree.right!=null){//如果有右子树
tree.right = getHead(tree.right);//那么根的right应该指向右子树形成双链表的头节点
tree.right.left = tree;//形成双链表
}
return head;//返回head
}
}