二叉搜索树的中序遍历得到的是一个已排序的序列。因此可以仿照中序遍历,依次访问结点并改变结点的指向,使其构成一个双向链表。在访问当前结点之前,需要先访问左子树,并改变左子树中的结点指向,使其构成双向链表。同时需要获取左子树构成双向链表的头和尾结点。头结点是整棵树构成的双向链表的头结点,而尾结点是左子树中最后一个访问的结点,用于和根结点连接。然后访问右子树,获取右子树链表的头结点和尾结点,头结点用于连接当前结点和右子树链表。所以执行的逻辑是:
1.若根节点为空,直接返回
2.根节点不为空,且左子树存在,先访问左子树,并获取左子树链表的头结点和尾结点,左子树的头结点即为整棵树构成链表的头结点,将左子树尾结点和根节点连接。
3.若左子树不存在,根就是链表的头结点
4.若右子树存在,访问右子树,获取到右子树链表的头结点。将根节点和右子树的头结点连接。
5.若右子树不存在,则根节点即为尾结点。
TreeNode* _convert(TreeNode* root, TreeNode** last) {
// 函数返回链表的头结点,last保存链表的尾结点
TreeNode* head, *mid;
if (root->left) {
head = _convert(root->left, last);
//将当前结点插入链表
(*last)->right = root;
root->left = *last;
} else {
head = root;
}
if (root->right) {
//将当前结点与有子树的链表相连
mid = _convert(root->right, last);
root->right = mid;
mid->left = root;
} else {
*last = root;
}
return head;
}
TreeNode* Convert(TreeNode* pRootOfTree)
{
if (pRootOfTree == nullptr)
return nullptr;
TreeNode* head, *last;
head = _convert(pRootOfTree, &last);
return head;
}
本文介绍了一种将二叉搜索树转换为双向链表的方法,通过中序遍历的方式,依次访问结点并改变结点指向,实现树到链表的转换。文章详细解释了递归算法步骤,包括处理空节点、左子树和右子树的情况。

1038

被折叠的 条评论
为什么被折叠?



