题目
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
输入输出示例:
思路
一开始想到的就是,因为二叉搜索树的中序遍历就是该二叉树的升序排列。所以先用一个list存储搜索树的中序遍历结果。然后根据这个list,对每个节点进行左右连接,即可得到双向链表。
第二种思路就是应用线索二叉树的原理。线索二叉树就是将二叉树中存在空指针的节点,让该节点的左指针指向前继节点,右指针指向后继节点,以此形成一个排序序列。但是这个过程中对于那些没有空指针的节点,它们与其连接的节点之间的连接是没有断开的,所以还是一颗树。
那么应用该原理,在节点时,直接把原有的连接破坏,那么树结构就会被破坏。然后进行左右连接,即可拿到链表。
具体思路就是像中序遍历一样,先向左递归,当pre(辅助指针)不为空时,就将该节点与pre进行连接,连接完后,将pre指向该节点,然后向右递归。用上述例子看的话,这样到最后可以得到链表,但是pre指向了7,输出就是逆序的。所以这里哦做了调整,先向右递归,连接,再向左递归。
代码
代码如下(示例):
使用list构建链表的代码
public class Solution {
public TreeNode Convert(TreeNode pRootOfTree) {
if(pRootOfTree==null){
return pRootOfTree;
}
ArrayList<TreeNode> list = new ArrayList<>();
in(pRootOfTree,list);
return buildList(list);
}
public void in(TreeNode root,ArrayList<TreeNode> list){
if(root!=null){
in(root.left,list);
list.add(root);
in(root.right,list);
}
}
public TreeNode buildList(ArrayList<TreeNode> list){
TreeNode node = list.get(0);
TreeNode temp = node;
for(int i = 1;i<list.size();i++){
temp.right = list.get(i);
list.get(i).left = temp;
temp = list.get(i);
}
return node;
}
}
使用list构建链表的代码
public class Solution {
TreeNode pre=null;
public TreeNode Convert(TreeNode pRootOfTree) {
if (pRootOfTree==null){
return null;
}
Convert(pRootOfTree.right);
if (pre!= null){
pRootOfTree.right=pre;
pre.left=pRootOfTree;
}
pre=pRootOfTree;
Convert(pRootOfTree.left);
return pre;
}
}