王道课后习题4.3.13:查找p和q的最近公共祖先结点r

TNode* Parent_Common(TNode* root,char p,char q)
{
    Stack S1;
    Stack S2;
    InitStack(S1);
    InitStack(S2);
    TNode* b1=root;
    TNode* b2=root;//不可以用同一个b,因为第一个while循环里b会改变。
    TNode* r=NULL;
    while(!StackEmpty(S1)||b1!=NULL)
    {
        if(b1!=NULL)
        {
            Push(S1,b1);
            b1=b1->lchild;
        }
        else
        {
            GetTop(S1,b1);
            if(b1->data==p)
                break;
            if(b1->rchild!=NULL&&b1->rchild!=r)
                b1=b1->rchild;
            else
            {
                Pop(S1,b1);
                r=b1;//标记右孩子已经访问过了
                b1=NULL;//访问完一个结点之后需要置空,不置空的话又会把这个结点压进去
            }
        }
    }
    while(!StackEmpty(S2)||b2!=NULL)
    {
        if(b2!=NULL)
        {
            Push(S2,b2);
            b2=b2->lchild;
        }
        else
        {
            GetTop(S2,b2);
            if(b2->data==q)
                break;
            if(b2->rchild!=NULL&&b2->rchild!=r)
                b2=b2->rchild;
            else
            {
                Pop(S2,b2);
                r=b2;//标记右孩子已经访问过了
                b2=NULL;//访问完一个结点之后需要置空,不置空的话又会把这个结点压进去
            }
        }
    }
    while(!StackEmpty(S1)&&!StackEmpty(S2))
    {
        TNode *p,*q;
        Pop(S1,p);
        Pop(S2,q);
        if(p->data==q->data)
            return p;
    }
    return NULL;
}

你可能感兴趣的:(树,王道课后习题+)