二叉搜索树的结点有2个指针,分别指向左右孩子,双链表的每个结点也有2个指针,分别指向前后结点,所以在不创建新结点,只调整指针指向时可以将二叉搜索树变成双向链表;又由于二叉搜索树在中序遍历时是有序的,所以可以采用中序处理二叉搜索树调整指针指向将其变成有序双向链表。为了简化指针移动操作,我们让左孩子为前向指针,右孩子为后向指针。
二叉搜索树的最左结点即使整个树中最小的结点,所以首先找到最左结点,它就是链表的首结点,链表最后一个结点初始化为空。中序遍历时,当前结点的左孩子指向链表的最后一个结点,若最后一个结点不为空,则最后结点的右孩子指向当前结点,当前结点作为最后一个结点。递归的在所有子树中进行。
#include<iostream>
using namespace std;
class tree_node{
public:
class tree_node* left,*right;
int data;
tree_node(int d,tree_node *l=NULL,tree_node *r=NULL)
{
data=d;
left=l;
right=r;
}
};
tree_node* insert_node(tree_node *root,int d)
{
tree_node *curr=root,*prev;
while(curr!=NULL)
{
prev=curr;
if(curr->data < d)
{
curr=curr->right;
}
else
{
curr=curr->left;
}
}
if(root==NULL)
{
root=new tree_node(d);
}
else
{
if(prev->data< d)
{
prev->right=new tree_node(d);
}
else
prev->left=new tree_node(d);
}
return root;
}
void PrintBSTree(tree_node* tree)
{
if(tree==NULL)
return;
PrintBSTree(tree->left);
cout << tree->data << " ";
PrintBSTree(tree->right);
}
tree_node * findLeft(tree_node *root )
{
if(root==NULL)
return NULL;
while (root->left!=NULL)
{
root=root->left;
}
return root;
}
void convertNode(tree_node *root,tree_node** lastnode)
{
if(root==NULL)
return;
if(root->left!=NULL)
convertNode(root->left,lastnode);
root->left=*lastnode;
if(*lastnode!=NULL)
(*lastnode)->right=root;
*lastnode=root;
if(root->right!=NULL)
convertNode(root->right,lastnode);
}
tree_node * treeTolist(tree_node *root,tree_node *head,tree_node *lastnode)
{
if(root==NULL)
return NULL;
head=findLeft(root);
lastnode=NULL;
convertNode(root,&lastnode);
return head;
}
void PrintDList(tree_node *h)
{
tree_node* curr=h;
while(curr!=NULL)
{
cout<<curr->data<<" ";
curr=curr->right;
}
}
int main()
{
tree_node* root=NULL;
tree_node* DList,*head=NULL,*tail=NULL;
for(int i=0;i<10;i++)
{
int d=rand()%100;
root=insert_node(root,d);
}
cout<<"二叉搜索树:"<<endl;
PrintBSTree(root);
head=treeTolist(root,head,tail);
cout<<endl<<"有序双向链表"<<endl;
PrintDList(head);
return 0;
}