树中两个结点的最低公共祖先

本文介绍了一种使用递归方法构建二叉树的方法,并实现了寻找两个节点最近公共祖先的功能。通过输入节点值的方式建立二叉树结构,并记录每个节点的父节点,最后利用栈来找出指定节点的路径并确定其最近公共祖先。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

#include"stdafx.h"
#include<stdio.h>
#include<stack>
#include<iostream>
using namespace std;


struct TREE{
    int v;
    TREE *left,*right;
}*rhead;


int shu[10009],all=0;
void bulid(TREE *head){
    int temp;
    scanf("%d",&temp);
    if(temp==0){
        head->left=NULL;
        head->right=NULL;
        head->v=0;
        return;
    }else{
        shu[temp]=1;
        head->v=temp;
        head->left=new TREE;
        bulid(head->left);
        head->right=new TREE;
        bulid(head->right);
    }
}


int Apre[10099],Bpre[10999];
int fa,fb,oka,okb;
void finda(TREE *head){
    
    if(oka==1)return;
    if(head->v==fa){
        oka=1;
        return;
    }
    
    if(head->left->v!=0){
        if(head->left->v!=head->v)
        Apre[head->left->v]=head->v;
        finda(head->left);
    }
    if(head->right->v!=0){
        if(head->right->v!=head->v)
        Apre[head->right->v]=head->v;
        finda(head->right);
    }
}


void findb(TREE *head){
    
    if(okb==1)return;
    if(head->v==fb){
        okb=1;
        return;
    }
    if(head->left->v!=0){
        if(head->left->v!=head->v)
        Bpre[head->left->v]=head->v;
        findb(head->left);
    }
    if(head->right->v!=0){
        if(head->right->v!=head->v)
        Bpre[head->right->v]=head->v;
        findb(head->right);
    }
}


void freeTree(TREE *head ){
    if(head->left!=NULL) freeTree(head->left);
    if(head->right!=NULL) freeTree(head->right);
    delete head;
    head=NULL;
}


int main(){
    int t;
    while(scanf("%d",&t)!=EOF)
    while(t--){
        int i;
        stack<int>sta,stb;


        for(i=0;i<=10000;i++){
            shu[i]=0;
            Apre[i]=-1;
            Bpre[i]=-1;
        }


        rhead=new TREE;
        TREE *head=rhead;
        bulid(head);
        oka=0;okb=0;
        scanf("%d%d",&fa,&fb);
        if(shu[fa]==0||shu[fb]==0){
            printf("My God\n");
        }else{
            finda(rhead);
            findb(rhead);


            int p=fa;
            sta.push(p);
            while(Apre[p]!=-1){
                p=Apre[p];
                sta.push(p);
                
            //    if(Apre[p]==p)break;
            }


            p=fb;
            stb.push(p);
            while(Bpre[p]!=-1){
                p=Bpre[p];
                stb.push(p);
                
            //    if(Bpre[p]==p)break;
            }


            int root=-1;
            while((!sta.empty())&&(!stb.empty())&&(sta.top()==stb.top())){
                root=sta.top();
                sta.pop();
                stb.pop();
            }
            if(root==-1)printf("My God\n");
            else 
            printf("%d\n",root);
            
            freeTree(rhead);
        }
    }


    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值