求出指定结点在给定二叉排序树中的层次
在二叉排序树中,查找一次就下降一层,因此,查找该结点所用的次数就是二叉排序树中的层次
int level(BiTree T, Bitnode *p)
{
int n = 0;
BiTree node = T;
if (node != NULL)
{
n++;
while (node->data != p->data)
{
if (p->data < node->data) // 在左子树中找
node = node->lchild;
else // 在右子树中找
node = node->rchild;
n++; // 层次+1
}
}
return n;
}
判断二叉树是否是平衡二叉树
采用后序遍历的思想
1.若为空树或只有根结点,则是平衡二叉树
2.否则对左右子树进行递归,树T的高度为最高子树的高度+1.若左右子树高度差大于1,则不是
若左右子树高度差小于或等于1时,且左右子树都平衡时,返回true
// 计算树的高度
int height(BiTree T)
{
if (T == NULL)
{
return 0;
}
int lchild_height = height(T->lchild);
int rchild_height = height(T->rchild);
return lchild_height > rchild_height ? lchild_height + 1 : rchild_height + 1;
}
// 判断是否为平衡二叉树
int isBalanced(BiTree T)
{
if (T == NULL)
{
return 1;
}
int lchild_height = height(T->lchild); // 左子树高度
int rchild_height = height(T->rchild); // 右子树高度
if (abs(lchild_height - rchild_height) <= 1 && isBalanced(T->lchild) && isBalanced(T->rchild))
{
return 1;
}
return 0;
}
从大到小输出所有值不小于K的关键字
void OutPut(BiTree T, int k)
{
if (T == NULL)
return;
if (T->rchild != NULL)
OutPut(T->rchild, k); // 递归输出右子树结点
if (T->data >= k)
printf("%d", T->data);
if (T->lchild != NULL) // 递归输出左子树结点
OutPut(T->lchild, k);
}
在一棵有n个结点的、随机建立起来的二叉排序树上查找第k小的元素,并返回指向该节点的指针,要求时间复杂度为O(logn)在一棵有n个结点的、随机建立起来的二叉排序树上查找第k小的元素,并返回指向该节点的指针,要求时间复杂度为O(logn)
核心思想是利用中序遍历,因为中序的结果是有序的
也可参考王道p286 11题代码
对于二叉排序树来说,中序遍历的结果就是树中元素按照从小到大的顺序排列的。我们可以利用这一特性来解决这个问题。
我们可以使用一个全局变量count来记录当前遍历到的结点数,使用一个全局变量result来记录第k小的元素。
对二叉排序树进行中序遍历,遍历到一个结点时,count加1,当count等于k时,说明找到了第k小的元素,此时将该结点的值赋给result,并返回。
递归遍历当前结点的左子树和右子树,直到找到第k小的元素。
// 全局变量,用于记录已经遍历过的结点数
int count = 0;
// 中序遍历二叉排序树,查找第k小的元素
void inorder(BiTree T, int k,BiTree result) {
if (T == NULL) {
return;
}
inorder(T->lchild, k,result);
count++; // 遍历到一个结点,计数器加1
if (count == k) { // 如果已经遍历到第k个结点
result = T; // 记录第k小的元素
return;
}
inorder(T->rchild, k,result);
}
BiTree kthSmallest(BiTree T, int k) {
count = 0; // 初始化计数器
BiTree result = NULL;
inorder(T, k, result); // 中序遍历二叉排序树
return result; // 返回第k小的元素
}