剑指offer面试题50!!!树中两个最低公共祖先

来源:互联网 发布:php v 和phpinfo版本 编辑:程序博客网 时间:2024/06/05 11:21

庆祝!!!最后一篇剑指offer系列博客!

/*这本书,,终于看到最后一个题了。太激动了。这还是以面试的实际案例给出的当给除这个题目的时候 首先看是不是二叉树如果是bst 那直接找比一个小 比另一个大的如果说只说普通的树 先问有没有指向父亲的指针 有的话就相当与 两个链表的第一个重合节点么如果没有指向父节点的指针 那也好办 直接遍历 然后记录路径就行了最后两个路径的最后一个相同的点就是ans了*/#include<iostream>#include<cstdio>#include<vector>#include<list>using namespace std;struct TreeNode{    int data;    vector<TreeNode *> child;    TreeNode(int t)    {        data=t;    }};//这里返回根节点的指针 然后传入两个要找的点的 指针的引用 第一次这么写 可以的 很好用TreeNode * CreateTree(TreeNode * &p1,TreeNode * &p2){    TreeNode * root=new TreeNode(1);    TreeNode * tmp=new TreeNode(2);    root->child.push_back(tmp);    tmp=new TreeNode(3);    root->child.push_back(tmp);    tmp=root->child[0];    TreeNode * tmp1=new TreeNode(4);    tmp->child.push_back(tmp1);    tmp1=new TreeNode(5);    tmp->child.push_back(tmp1);    TreeNode * tmp2=tmp->child[1];//保存5    tmp=tmp->child[0];    tmp1=new TreeNode(6);    p1=tmp1;    tmp->child.push_back(tmp1);    tmp1=new TreeNode(7);    tmp->child.push_back(tmp1);    tmp=tmp2;    tmp1=new TreeNode(8);    p2=tmp1;    tmp->child.push_back(tmp1);    tmp1=new TreeNode(9);    tmp->child.push_back(tmp1);    tmp1=new TreeNode(10);    tmp->child.push_back(tmp1);    return root;}void ShowTree(TreeNode * root){    printf("%d\t",root->data);    for(int i=0;i<root->child.size();++i)        ShowTree(root->child[i]);}bool GetNodePath(TreeNode * root,TreeNode *p,list<TreeNode *> & path){    if(root==p)// 找到这个点了 返回    return true;    path.push_back(root);    bool find=false;    vector<TreeNode *> ::iterator it=root->child.begin();    while(!find && it!=root->child.end())    {        find=GetNodePath(*it,p,path);        ++it;    }    if(!find)//没找到就退一下    path.pop_back();    return find;}TreeNode * GetLastCommonParend(TreeNode * root, TreeNode * p1, TreeNode * p2){    if(root==NULL || p1==NULL || p2==NULL)    return NULL;    list<TreeNode * > path1;//记录根节点到两个节点的路径    list<TreeNode * > path2;    GetNodePath(root,p1,path1);//调用这个函数 最后 path1 和path2 里面存了路径    GetNodePath(root,p2,path2);    if(path1.empty() || path2.empty())//说明有最少一个点 就是root    return root;    //接下来就把这两个路径中最后一个相等的元素找出来就行了    TreeNode * plast=NULL;    list<TreeNode *>::iterator it1=path1.begin();    list<TreeNode *>::iterator it2=path2.begin();    while(it1!=path1.end() && it2!=path2.end())    {        if(*it1==*it2)//这里只要地址一样就可以了            plast=*it1;        it1++;        it2++;    }    return plast;}int main(){    TreeNode * p1,* p2;    TreeNode * root=CreateTree(p1,p2);    //到这里 root p1 p2 就弄好了 现在要得到p1 p2的最低公共祖先   // ShowTree(root);    TreeNode * ans=GetLastCommonParend(root,p1,p2);    if(ans!=NULL)    cout<<ans->data<<endl;    return 0;}


原创粉丝点击