剑指offer题解（C语言）----JZ08 二叉树的下一个结点

题目

给定一个二叉树其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点的next指针。下图为一棵有9个节点的二叉树。树中从父节点指向子节点的指针用实线表示，从子节点指向父节点的用虚线表示

示例:

输入:{8,6,10,5,7,9,11},8

返回:9

数据范围：1≤n≤50  ，节点上的值满足 1≤val≤100 

要求：空间复杂度 O(1)  ，时间复杂度O(n) 

输入描述：

输入分为2段，第一段是整体的二叉树，第二段是给定二叉树节点的值，后台会将这2个参数组装为一个二叉树局部的子树传入到函数GetNext里面，用户得到的输入只有一个子树根节点

返回值描述：

返回传入的子树根节点的下一个节点，后台会打印输出这个节点

解析

其实拿到题的第一反应是：很像二叉查找树的中序遍历嘛！第二想法是：有点像线索二叉树！

后来看了看其他示例觉得都不对。线索二叉树是帮助找到结点前驱的，而不是找后继。本题还是老老实实找规律，分析清楚后继结点是如何产生的更为靠谱。

找规律的过程确实有点难...

（下图源知乎水印）

根据上图，红虚线所指向的就是每个结点的后继，不难发现以下规律：

1. 某结点有右子树(D,B,E,A,C,G)：其后继结点为右子树最左边的点；

2. 某结点没有右子树，但是是其父节点的左孩子(N,I,L)：其后继结点就是其父节点；

3. 某结点没有右子树，且不是父节点的左孩子(H,J,K,M)：其后继节点为其父节点的父节点的...直到当前结点为其父节点的左孩子为止；

4. 如果上述情况均不满足，则其后继结点为空，返回NULL；

本题需要注意的是：next指针指向的是当前结点的父节点，而非字面意思的“下一个/后继”。

代码

struct TreeLinkNode* GetNext(struct TreeLinkNode* pNode ) {
    struct TreeLinkNode* tmp;
    if(!pNode)return pNode;
    if(pNode->right == NULL){ // 如果pNode没有右子树
        while(pNode->next){ 
            //将当前结点父节点的值赋给tmp
            tmp = pNode->next; 
            //如果当前结点是父节点的左孩子，则返回父节点的值
            if(pNode == tmp->left) return pNode->next; 
            //如果当前结点不是左孩子，则一直寻找其父节点，直到是左孩子为止
            pNode = pNode->next;
        }
        return NULL;