考研数据结构(每日一题)
题目:假设二叉树采用二叉链表存储结构,设计一个算法求二叉树的高度(递归和非递归)
算法思想:
递归:递归求左子树高度、右子树高度,取较大者+1
非递归:采用层次遍历算法,变量level记录二叉树层数,指针last指向最右结点,每次层次遍历出队时与last指针比较,若两者相等,则层数加1,并让last指向下一层的最右结点,直到遍历完成。level的值即为二叉树的高度。
完整代码:
//递归
int Btdepth(Bitree T){
if(T == NULL){ //空树高度为0
return 0;
}
ldep = Btdepth(T -> lchild); //左子树高度
rdep = Btdepth(T -> rchild); //右子树高度
if(ldep > rdep){
return ldep + 1;
}else{
return rdep + 1;
}
}
//非递归
int Btdepth(BiTree T){
if(!T){
return 0; //树空则高度为0
}
int front = -1,rear = -1;
int last = 0,level = 0; //last指向当前层的最右结点
BiTree Q[MaxSize]; //设置队列Q,元素是二叉树结点指针且容量足够
Q[++rear] = T; //将根节点入队
BiTree p;
while(front < rear){ //队列不空,循环
p = Q[++front]; //队列元素入队,访问结点
if(p -> lchild){
Q[++rear] = p -> lchild; //左孩子入队
}
if(p -> rchild){
Q[++rear] = p -> rchild; //右孩子入队
}
if(front == last){ //处理该层的最右结点
level ++; //层数加1
last = rear; //last指向下层
}
}
return level;
}