2017百度面试现场coding算法三

struct TreeNode
{
int value;
TreeNode* first_child;
TreeNode* next_sibling;
TreeNode(int a):value(a),first_child(NULL),next_sibling(NULL){}
};
bool hasNode(TreeNode* pNode,TreeNode* p)
//判断以pNode为根的树中有没有节点
{
bool flag1=false,flag2=false;
if(pNode==NULL) return false;
if(pNode==p)return true;//当前节点即是访问节点
else
{
if(pNode->first_child!=NULL)
{flag1=hasNode(pNode->first_child, p);//节点在孩子节点
TreeNode* p1=pNode->first_child;
while(p1->next_sibling!=NULL)//在同层兄弟节点中？
{
if(hasNode(p1->next_sibling, p))//找到
{   flag2=true;
break;
}
p1=p1->next_sibling;//循环查找是否在同层的其他节点
}
}
}
return flag1||flag2;//只要包含在孩子或者孩子的兄弟节点中
}
TreeNode* pubParent(TreeNode* root,TreeNode* p,TreeNode* q)
{
TreeNode* pNode=root;
while(hasNode(pNode, p)&&hasNode(pNode, q))//当前节点包含两个节点
{
bool flag1=false,flag2=false;
if(pNode->first_child!=NULL&&hasNode(pNode->first_child, p)&&hasNode(pNode->first_child, q))
//判断是否在孩子节点所在的树中
{    pNode=pNode->first_child,flag1=true;}
else if(pNode->first_child->next_sibling!=NULL)//判断是否在孩子同层的某个兄弟树中
{
TreeNode* p1=pNode->first_child;
while(p1->next_sibling!=NULL)
{
if(hasNode(p1->next_sibling, p)&&hasNode(p->next_sibling, q))//找到包含的兄弟树
{    pNode=p1->next_sibling;
//继续以兄弟树为根查找
flag2=true;
break;
}
else
p1=p1->next_sibling;
}
if(flag2==true)//当包含在其中之一的兄弟节点，则以该节点为根，继续查找
break;
}
if(flag1==false&&flag2==false)//若没有在孩子节点和孩子的兄弟节点中，则只在当前节点中，返回当前节点
return pNode;
}
return pNode;
}