建立二叉搜索树的过程,就是将一个一个数插入数集的过程,只不过数集的结构是一棵二叉树。红黑树就是一棵平衡二叉搜索树。
插入一个数,开始将该数与树中的结点比较,从根节点开始。如果根结点就是空的,直接将该数放在根的位置;否则,与根节点的值相比较,比其小则该数肯定插在根节点的左子树上,比其大则该数肯定插在根节点的右子树上。
以以上算法为依据写出插入函数的代码如下(color代表节点的颜色,true为黑,false为红):
//建树插入函数
node* Insert(node *root,int v,bool color)
{
if(root==NULL)
{
root=new node;
root->val=v;
root->isBlack=color;
root->left=NULL;
root->right=NULL;
}
else
{
if(v<root->val)
{
//继续往左子树插
root->left=Insert(root->left,v,color);
}
else if(v>root->val)
{
//继续往右子树插
root->right=Insert(root->right,v,color);
}
}
return root;
}
调用Insert函数部分的代码(正数表示黑节点,负数表示红节点):
for(i=0;i<n;i++)
{
scanf("%d",&temp);
bool isBlack=true;
if(temp<0)
{
temp=-temp;
isBlack=false;
}
root=Insert(root,temp,isBlack);
}
//建树完成