面试题58：二叉树的下一个结点

这个题是在《剑指offer》的英文版新增的面试题，但今天花了我大半天的时间去理解。

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

struct TreeLinkNode {    int val;    struct TreeLinkNode *left;    struct TreeLinkNode *right;    struct TreeLinkNode *next;    TreeLinkNode(int x) :val(x), left(NULL), right(NULL), next(NULL) {    }};

这是牛客给出的结构，讲真最开始一直很懵逼，这个next是啥，最后才明白是parent的意思。因为这个树有一个结点是指向父亲的。那么下面我们来分析这个问题。

首先这是一棵按照要求构造的树。假如我们要找的是4中序遍历的下一个结点，毫无疑问是返回next就OK了。如果是2的下一个结点呢？这里就有点复杂了，因为在中序遍历下，先遍历的是右子树的最左结点，也就是图中的8。在假如当前结点是9，那么下一个结点是1，也就是说在这种情况下，当前结点是根的右子树且没有右子树，那么我们就需要向上寻找，找到某个结点，满足父节点是它的左子树这个要求。对7这个结点来说也是一样的，我们寻找某个结点，但是1没有父节点了，所以g没有next结点。情况分类如下。

说完可能出现的情况，我们直接根据情况上代码。

class Solution {public:    TreeLinkNode* GetNext(TreeLinkNode* pNode)    {        if(pNode==NULL)            return NULL;        TreeLinkNode* nextnode=NULL;        TreeLinkNode* cur=pNode;        if(cur->right!=NULL)   //找2的下一个        {            TreeLinkNode* RightNode=cur->right;            while(RightNode->left!=NULL)                RightNode=RightNode->left;            nextnode=RightNode;        }        else if(cur->next!=NULL)   //  4和9的情况        {            TreeLinkNode* parent=cur->next;            while(parent!=NULL && cur==parent->right)  //找9的下一个            {                cur=parent;               parent=parent->next;            }            nextnode=parent;        }        return nextnode;    }};

其实这道题分情况讨论并不难，但是在开始的理解题意花了很长的时间。但对于这种情况比较多的题目，先将问题理解明白，然后根据每种情况去讨论问题就会迎刃而解。