《剑指offer》面试题8：二叉树的下一个结点（附建树过程）

题目

给定一个二叉树和结点指针，输出这个结点对应中序遍历的下一个结点。二叉树结点中除了有左右孩子指针，还有双亲指针。

分析

我开始就思考了双亲结点的指针是如何得到的，其实只要用到层次遍历去遍历每个结点的时候，刚好要判断是否有左右孩子结点，可以在这时对左右孩子结点注入双亲结点指针。这样就可以在一次遍历之后知道所有结点的双亲。

void BFS(Node *node){
	queue<Node*> q;
	q.push(node);
	int num=0;
	while(!q.empty()){
		Node* temp = q.front();
		q.pop();
		num++;
		if(num < 8) printf(" ");  //这个8是我虚构的，其实就是结点总数
		if(temp->lchild != NULL){
            q.push(temp->lchild);
            temp->lchild->parent = temp;//注入双亲的关键就在这一步
		}
		if(temp->rchild != NULL){
                q.push(temp->rchild);
                temp->rchild->parent = temp;//同上
		}
	}
}

判断一个结点的中序的下一个结点要分类讨论：

1. 结点有右子树，那么下一个结点应该是左右树下最左边的那个结点。
2. 结点没有右子树，相当于遍历完它之后，会往上找双亲结点，此时又要分类：要么双亲为空（只有根才会没有双亲），此时不存在下一个结点；要么双亲不会空，且它刚好是双亲的左孩子，此时下一个结点就是双亲；要么双亲不会空，且它是双亲的右孩子，这时代表以双亲为根的这颗树已经遍历完，所以要找双亲的双亲，然后进行和上面一样的判断。

实现

Node* FindNextNode(Node* root){
    Node* ans=NULL;//用ans来接收结果
    if(root==NULL){
        return ans;
    }
    if(root->rchild!=NULL){    //右子树不为空最简单
        ans = root->rchild;
        while(ans->lchild!=NULL){
            ans = ans->lchild;
        }
    }else if(root->parent != NULL){  //右子树为空，但是双亲不为空
        Node* p = root->parent;
        Node* now = root;
        while(p!=NULL && p->rchild == now){   
            now = p;            //记录下此时需要被查找双亲的结点，因为如果不符合条件，需要找双亲的双亲，如果还不符合，再找双亲的双亲，知道跳出循环
            p = p->parent;
        }
        ans = p;
    }
    return ans;   //右子树为空，双亲也为空，此时直接返回ans的默认值NULL
}

思考

如果没有用parent指针，还可以用线索树（中序），很容易找到下一个结点，不过结点的数据结构需要添加两个标记，用来标识rchild是右结点还是后驱结点。