题目描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
思路 & 解答
主要是使用递归,设置一个head
和end
,head
只在第一次设置值,后面就是将end
的right
指向当前node
,将当前node
的left
指向end
,把end
指向当前的node
。
代码如下:
class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
public class Solution {
public TreeNode head = null;
public TreeNode end = null;
public TreeNode Convert(TreeNode pRootOfTree) {
invert(pRootOfTree);
return head;
}
public void invert(TreeNode node) {
if (node == null) {
return;
}
invert(node.left);
if (end == null) {
head = node;
end =node;
}else{
end.right = node;
node.left = end;
end =node;
}
invert(node.right);
}
}
C++
代码如下:
class Solution {
public:
TreeNode *head;
TreeNode *end;
TreeNode *Convert(TreeNode *pRootOfTree) {
invert(pRootOfTree);
return head;
}
void invert(TreeNode *node) {
if (node == NULL) {
return;
}
invert(node->left);
if (end == NULL) {
head = node;
end = node;
} else {
end->right = node;
node->left = end;
end = node;
}
invert(node->right);
}
};
时间复杂度:O(N)。 空间复杂度:O(N),主要是递归的栈空间。