剑指offer：二叉树的下一节点

题目描述

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

思路：

画一个三层的满二叉树，每个节点分析一遍找到规律。

有三种情况：

1. 给定节点有右子树，则下一节点是右子树的最左叶节点；

2. 给定节点没有右子树，如果给定节点是其父节点的左孩子，则下一节点是父节点；

3. 给定节点没有右子树，并且给定节点是其父节点的右孩子，则向上一直找，直到找到一个节点p，它是其父节点的左孩子。如果找不到p，说明给定节点是最后一个节点了，如果找到了p，则下一节点是p的父节点。

参考代码：

在线测试

https://www.nowcoder.com/practice/9023a0c988684a53960365b889ceaf5e?tpId=13&tqId=11210&rp=3&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

AC代码

/*
struct TreeLinkNode {
    int val;
    struct TreeLinkNode *left;
    struct TreeLinkNode *right;
    struct TreeLinkNode *next;
    TreeLinkNode(int x) :val(x), left(NULL), right(NULL), next(NULL) {
        
    }
};
*/
class Solution {
public:
    TreeLinkNode* GetNext(TreeLinkNode* pNode)
    {
        if(pNode==NULL)
		return NULL;
	    if(pNode->right!=NULL)
	    {
		    TreeLinkNode* temp=pNode->right;
		    while(temp->left!=NULL)
		    {
		    	temp=temp->left;
		    }
		    return temp;
	    }
	    else
	    {
	    	if(pNode->next==NULL)
	    		return NULL;
	    	if(pNode==pNode->next->left)
	    		return pNode->next;
	    	else
		    {
		    	TreeLinkNode* temp=pNode->next;
		    	while(temp!=NULL&&temp->next!=NULL&&temp!=temp->next->left)
                {
		    		temp=temp->next;
		    	}
		    	return temp->next;//注意这里是返回找到的p的父节点，如果走到了根节点，没有父节点，返回的也是NULL
		    }
	    }
    }
};