题目描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
思路:递归
1.将左子树构造成双链表,并返回链表头节点。
2.定位至左子树双链表最后一个节点。
3.如果左子树链表不为空的话,将当前pRootOfTree追加到左子树链表。
4.将右子树构造成双链表,并返回链表头节点。
5.如果右子树链表不为空的话,将该链表追加到pRootOfTree节点之后。
6.根据左子树链表是否为空确定返回的节点。
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def Convert(self, pRootOfTree):
# write code here
if pRootOfTree is None://根为空
return None
if pRootOfTree.left==None and pRootOfTree.right==None://左右子树为空,则返回根
return pRootOfTree
lefttree=self.Convert(pRootOfTree.left)//左子树构成双链表
p=lefttree//定义一个临时p,用来定位左子树最后一个节点
while(p and p.right)://定位左子树最后一个节点
p=p.right
if(lefttree)://将左子树最后一个节点连接到根
p.right=pRootOfTree
pRootOfTree.left=p
righttree=self.Convert(pRootOfTree.right)//右子树构造双链表
if(righttree)://右子树追加到根节点之后
righttree.left=pRootOfTree
pRootOfTree.right=righttree
if lefttree://左子树,返回左子树
return lefttree
else:
return pRootOfTree//否则,返回根
/**
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;
if(pRootOfTree.left==null && pRootOfTree.right==null) return pRootOfTree;
TreeNode lefttree=Convert(pRootOfTree.left);
TreeNode p=lefttree;
while(p!=null && p.right!=null) p=p.right;
if(lefttree!=null){
p.right=pRootOfTree;
pRootOfTree.left=p;
}
TreeNode righttree=Convert(pRootOfTree.right);
if(righttree!=null){
righttree.left=pRootOfTree;
pRootOfTree.right=righttree;
}
return lefttree==null?pRootOfTree:lefttree;
}
}