【剑指Offer面试编程题】题目1509:树中两个结点的最低公共祖先--九度OJ

论坛 期权论坛 编程之家     
选择匿名的用户   2021-6-2 20:53   1376   0
题目描述:

给定一棵树,同时给出树中的两个结点,求它们的最低公共祖先。

输入:

输入可能包含多个测试样例。
对于每个测试案例,输入的第一行为一个数n(0<n<1000),代表测试样例的个数。
其中每个测试样例包括两行,第一行为一个二叉树的先序遍历序列,其中左右子树若为空则用0代替,其中二叉树的结点个数node_num<10000。
第二行为树中的两个结点的值m1与m2(0<m1,m2<10000)。

输出:

对应每个测试案例,
输出给定的树中两个结点的最低公共祖先结点的值,若两个给定结点无最低公共祖先,则输出“My God”。

样例输入:

2
1 2 4 6 0 0 7 0 0 5 8 0 0 9 0 0 3 0 0
6 8
1 2 4 6 0 0 7 0 0 5 8 0 0 9 0 0 3 0 0
6 12
样例输出:

2
My God
【解题思路】首先是树的输入,这个利用中序遍历的特点递归输入就行。然后,我们需要判断给出的两个节点是否都在树中,不在直接输出非法信息。若都在树中则继续查找,这个时候我们需要递归查找,给定一个头节点和两个待查的节点值,首先判断如果节点值等于头结点的节点值,那么公共节点肯定就是头节点;否则,继续在头结点的左支和右支中间查找,若两个节点值分居左右支树中,那么公共节点肯定也就是头结点。否则我们可以通过左支或者右支返回的节点得到公共节点。

注意根节点需要通过传地址调用才能返回值,所以create参数是指针的指针。

AC code:

#include <cstdio>
#include <set>
using namespace std;
 
struct nod
{
  int val;
  nod *lc,*rc;
};
 
set<int> setin;
 
void create(nod **nd)
{
  int tt;
  scanf("%d",&tt);
  if(tt==0) {nd=NULL; return;}
  else
  {
    setin.insert(tt);
    nod *newnd=new nod();
    newnd->val=tt;
    *nd=newnd;
    create(&(newnd->lc));
    create(&(newnd->rc));
  }
}
 
nod * lca(nod *nd,const int &n,const int &m)
{
  if(!nd) return NULL;
  if(nd->val==n || nd->val==m) return nd;
  nod *lnd=lca(nd->lc,n,m);
  nod *rnd=lca(nd->rc,n,m);
  if(lnd && rnd) return nd;
  return lnd?lnd:rnd;
}
 
int main()
{
  int n;
  scanf("%d",&n);
  for(int i=0;i<n;++i)
  {
    nod *head=NULL;
    setin.clear();
    create(&head);
    int n,m;
    scanf("%d%d",&n,&m);
    if(setin.find(n)==setin.end()  || setin.find(m)==setin.end())
    {
      printf("My God\n");
      continue;
    }  
    nod *re=lca(head,n,m);
    printf("%d\n",re->val);
  }
  return 0;
}
/**************************************************************
    Problem: 1509
    User: huo_yao
    Language: C++
    Result: Accepted
    Time:160 ms
    Memory:4988 kb
****************************************************************/

题目链接:http://ac.jobdu.com/problem.php?pid=1509

九度-剑指Offer习题全套答案下载:http://download.csdn.net/detail/huoyaotl123/8276299



分享到 :
0 人收藏
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

积分:3875789
帖子:775174
精华:0
期权论坛 期权论坛
发布
内容

下载期权论坛手机APP