题目:
假设二叉树采用二叉链表方式存储, root指向根结点,p所指结点和q所指结点为二叉树中的两个结点,编写一个计算它们的最近的共同祖先,函数定义如下:
BiTNode * nearest_ancestor(BiTree root, BiTNode *p, BiTNode *q);
其中 root 指向二叉树的根结点,p 和 q 分别指向二叉树中的两个结点。
path函数及栈相关定义如下:
bool path(BiTNode* root, BiTNode* node, Stack* s);
#define Stack_Size 50
typedef BiTNode* ElemType;
typedef struct{
ElemType elem[Stack_Size];
int top;
}Stack;
void init_stack(Stack *S); // 初始化栈
bool push(Stack* S, ElemType x); //x 入栈
bool pop(Stack* S, ElemType *px); //出栈,元素保存到px所指的单元,函数返回true,栈为空时返回 false
bool top(Stack* S, ElemType *px); //获取栈顶元素,将其保存到px所指的单元,函数返回true,栈满时返回 false
bool is_empty(Stack* S); // 栈为空时返回 true,否则返回 false
提示:在完成本题时,可利用 path 函数获取p和q两个结点到根结点之间的路径,之后再计算两条公共路径得出最近的共同祖先。
可以直接用path函数就很简单,题目已经给提示了。😊
代码:
#include <stdlib.h>
#include <stdio.h>
#include "bitree.h" //请不要删除,否则检查不通过
BiTNode * nearest_ancestor(BiTree root, BiTNode *p, BiTNode *q){
// 分别用栈Sp和Sq记录从根节点到节点p和q的路径
Stack Sp, Sq;
init_stack(&Sp);
init_stack(&Sq);
path(root, p, &Sp);
path(root, q, &Sq);
// 由于公共祖先一定在同一层,所以先让两个栈的栈顶在同一层
while (Sp.top != Sq.top) {
if (Sq.top > Sp.top) {
Sq.top--;
}
else {
Sp.top--;
}
}
// 依次弹出栈顶元素进行比较,找到最近公共祖先
while (!is_empty(&Sq)) {
BiTNode* pop_q = NULL;
pop(&Sq, &pop_q);
BiTNode* pop_p = NULL;
pop(&Sp, &pop_p);
if (pop_p == pop_q) {
// 如果栈顶元素相等,则说明找到了最近公共祖先
return pop_q;
}
}
// 如果没有找到最近公共祖先,则返回NULL
return NULL;
}