如果对数据结构不够了解,可以看看: 万字长文解说数据结构
36.二叉搜索树和双向链表
题目描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
思路 & 解答
主要是使用递归,设置一个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),主要是递归的栈空间。
【作者简介】:
秦怀,公众号【秦怀杂货店】作者,个人网站:aphysia.cn,技术之路不在一时,山高水长,纵使缓慢,驰而不息。