题目:
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
思路:
对于二叉树,只要想到有序,那么我们直接想到中序遍历。
中序遍历为LVR。此题让我们返回有序链表的头结点。如果我们使用LVR,并且L返回的是头结点,那么我们需要遍历找到尾结点,然后将当前节点插入。
RVL就不需要进行遍历,R每次返回的都是头结点,我们直接将当前节点头插到头节点,然后更新头节点就好。
代码实现:
RVL:
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
private TreeNode head=null;
public TreeNode Convert(TreeNode pRootOfTree) {
if(pRootOfTree==null)
return null;
Convert(pRootOfTree.right);
if(head==null){
head=pRootOfTree;
}else{
head.left=pRootOfTree;
pRootOfTree.right=head;
head=pRootOfTree;
}
Convert(pRootOfTree.left);
return head;
}
}
LVR:
/**
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 null;
//left为左子树链表头结点
TreeNode left=Convert(pRootOfTree.left);
TreeNode p=left;
//遍历找到尾结点,将当前节点链接进去
while(p!=null&&p.right!=null)
p=p.right;
if(left!=null){
p.right=pRootOfTree;
pRootOfTree.left=p;
}
//右子树返回的是头节点,直接将头结点与当前节点进行链接。
TreeNode right=Convert(pRootOfTree.right);
if(right!=null){
right.left=pRootOfTree;
pRootOfTree.right=right;
}
//返回当前有序链表的头结点。
return left==null?pRootOfTree:left;
}
}