将二叉查找树转换为链表
二叉查找树在不创建任何新节点的条件下将二叉查找树转换为排序的双向链表。
solution_1
二叉查找树的特点就是左子节点<根节点<右子节点
所以,想办法用中序遍历调整一下指针的指向就可以了
我们有一个树的根节点,然后假设还有一个链表的尾结点(因为我们从最小的节点开始,所以是往链表的后面加新的节点)
然后,根据我们的思路,我们要从树的最底层开始往上,并且是中序遍历
树里面最常用的就是递归
因此我们先找到最底层的左子树(如果有的话)
if (root->left)
transform(pcurrent->left,plastListNode);
这样一直到pcurrent->left->left
是NULL为止,这个时候pcurrent->left
是最后一个非空节点。
然后我们进行一些修改指针指向的工作:
pcurrent->left = plastListNode;
if (plastListNode)
{
plastListNode->right = pcurrent;
}
plastListNode = pcurrent;
大家可以自己演练一下怎么指指针。。
然后在后面,由于是中序遍历,所以加上右边的就可以了
#include <iostream>
struct treeNode
{
int data;
treeNode* left;
treeNode* right;
};
void transform(treeNode* root,treeNode*& plastListNode)
{
if (!root)
return;
treeNode* pcurrent=root;
if (root->left)
transform(pcurrent->left,plastListNode);
pcurrent->left = plastListNode;
if (plastListNode)
{
plastListNode->right = pcurrent;
}
plastListNode = pcurrent;
if (pcurrent->right)
transform(pcurrent->right, plastListNode);
}
treeNode* solution1(treeNode* tree)
{
treeNode* list = NULL;
transform(tree, list);
treeNode* headOfList = list;
while (headOfList&&headOfList->left)//转换到表头
headOfList = headOfList->left;
return headOfList;
}
solution_2
第二种解法呢,和我们一般的思维比较像,分左边和右边两种情况调整指针指向。
if (tree->left) pleft = transform_2(tree->left, false); //now pleft is the last & left of tree if (pleft) { pleft->right = tree; //tree->left = pleft; }
这段代码就是先找到最左边的节点,然后让pleft的右指针指向其父节点。
其实我觉得//tree->left = pleft;
这句话是不需要的,因为本来tree就是pleft的父节点,我把这行代码注释掉试了一下也是可以的。
然后后面就是同样的对最下面的右节点进行调整。
调整完了之后,为了之后的工作,我们把指针的位置调整到链表的头或者尾。
取决于我们当前调整的节点是左边的还是右边的。每次调整结束后,我们的指针总是指向父节点:
- 如果调整的是左边的,那么调整结束后,我们把指针右移到最右端,方便接着;
- 如果是右边的,我们把指针移到链表的最开始,方便打印。
。。其实我还是没懂为啥要这样调整。(主要是递归的执行过程)
我写了一个打印的程序试了一下:
执行结果这个还需要再理解。。
code:
treeNode* transform_2(treeNode* tree, bool asright)
{
if (!tree)
return NULL;
treeNode* tmp=tree;
treeNode* pleft=NULL;//what?!!!-ver3.0
treeNode* pright=NULL;
if (tmp->left)
pleft = transform_2(tmp->left, false);
//now pleft is the last & left of tree
if (pleft)
{
pleft->right = tmp;
//tree->left = pleft;
std::cout << "connectleft" << "\n";
}
if (tmp->right)
pright = transform_2(tmp->right, true);
if (pright)
{
pright->left = tmp;
//tree->right = pright;
std::cout << "connectright" << "\n";
}
treeNode* phead = tree;
if (asright)
{
while (phead->left)
phead = phead->left;
std::cout << "asright" << "\n";
}
else
{
while (phead->right)
{
phead = phead->right;
}
std::cout << "asleft" << "\n";
}
return phead;
}
文章参考何海涛大神文章