《剑指offer》c++版本 8.二叉树的下一个结点

如题：

给定一个二叉树和其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点
的指针。

这道题目考察对二叉树中序遍历的理解，做题的时候，如果没有思路，简单画画图，可以帮助理解。

上图没有画出指向父节点的指针，不过，不妨碍分析问题。已知节点，求中序遍历后一个节点。

如果节点存在右子树，显然，后一个节点位于右子树上最左边位置，递归查找即可。

如果没有右子树，且为根节点，显然，这时候没有下一个节点。

如果存在父节点，分为两种情况，如果是父节点的左孩子，下一个节点就是父节点。

如果是父节点的右孩子，这时候，继续顺着父节点指针网上搜索，直到找到一个位于左孩子的父节点，如上图中的19，是20的左孩子，8的后一个节点即是20。反之，若是，找不到，则没有。如上图右下角的7，父节点都位于右孩子。

经过上述分析可知，总共4种情况，依次处理即可。这道题如果没有思路，一定要画图。

下面是本题的c++解法：

/*
struct TreeLinkNode {
    int val;
    struct TreeLinkNode *left;
    struct TreeLinkNode *right;
    struct TreeLinkNode *next;
    TreeLinkNode(int x) :val(x), left(NULL), right(NULL), next(NULL) {
        
    }
};
*/
//这道题画图结合图形可以分析仅存在三种情况
//1. 节点存在右子树，此时，找到右子树最左节点即可。
//2. 没有父节点可直接返回。
//3. 节点不存在右子树且其存在父节点，且为父节点的左孩子，此时，返回父节点.
//4. 节点为父节点右子树，此时，递归向上查找，找到为左孩子的父节点，返回即可。
class Solution {
public:
    TreeLinkNode* GetNext(TreeLinkNode* pNode)
    {
        //特殊情况处理
        if (pNode == nullptr)
            return nullptr;
        
        //1. 存在右子树，找到最左节点
        if (pNode->right)
        {
            pNode = pNode->right;
            while(pNode->left)
                pNode = pNode->left;
            return pNode;
        }
        
        //2. 不存在右子树，且为根节点
        if (pNode->next == nullptr)
            return nullptr;
        
        //3. 存在父节点，且为父节点的左孩子
        if (pNode->next->left == pNode)
            return pNode->next;
        
        //4. 存在父节点，且为父节点的右孩子
        pNode = pNode->next;
        while (pNode && pNode->next)
        {
            if (pNode == pNode->next->left)
                return pNode->next;
            else
                pNode = pNode->next;
        }
        return nullptr;
    }
};

=============================================================================================

Linux应用程序、内核、驱动、后台开发交流讨论群(745510310)，感兴趣的同学可以加群讨论、交流、资料查找等，前进的道路上，你不是一个人奥^_^。