题目描述:
给定一棵二叉树和一个结点,要求找到中序遍历此树时该结点的下一个结点。
分析:
中序遍历一棵二叉树时,要根据一个结点有无右子树而分开讨论。
- 若当前结点有右子树,则它的下一个结点就是其右子树的最左叶子结点;
- 若当前结点没有右子树,那么又分两种情况:
*若当前结点是其父节点的左孩子结点,那么其下一个结点就是其父结点;
*若当前结点是其父节点的有孩子结点,那么沿着指向其父结点的指针向上查找,直到找到一个结点A满足A = A->parent->left 时,则下一个结点就是A->parent ;若一直找到根结点都没有满足要求的结点,那么当前结点没有下一个结点;
struct Node{
int val;
Node *left;
Node *right;
Node *parent;
Node(int x = 0):val(x), left(NULL), right(NULL), parent(NULL){}
};
Node* GetNextNode(Node *pNode){
if(pNode == NULL)
return NULL;
Node *dest = NULL;
if(pNode->right != NULL){
Node *pRight = pNode->right;
while(pRight->left != NULL)
pRight = pRoght->left;
dest = pRight;
}else{
Node *pCur = pNode;
Node *par = pNode->parent;
while(par != NULL pCur != par->left){
pCur = par;
par = par->parent;
}
dest = par;
}
return dest;
}