剑指offer——二叉树的下一个节点（c++)

题目描述

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

思路

图中的二叉树的中序遍历序列是{d, b, h, e, i, a, f, c, g}，我们以这棵树为例来分析如何找出二叉树的下一个结点。

如果一个结点有右子树，那么它的下一个结点就是右子树的最左子结点。例如，图中结点b的下一个结点是h，结点a的下一个结点是f。
如果一个结点没有右子树，且该结点是其父结点的左子结点，那么它的下一个结点就是其父结点。例如，图中结点d的下一个结点是b，结点f的下一个结点是c。
如果结点是其父结点的右子结点，可以沿着指向父结点的指针一直向上遍历，直到找到一个是它父结点的左子结点的结点，如果这样的结点存在，那么这个结点的父结点就是我们要找的下一个结点。例如，为了找到结点i的下一个结点，我们沿着指向父结点的指针，到达结点e。由于结点e是他父结点b的右结点，我们继续向上遍历到达结点b。结点b是它父节点a的左子结点。因此，结点a就是结点i的下一个结点。

struct TreeLinkNode {
    int val;
    struct TreeLinkNode *left;
    struct TreeLinkNode *right;
    struct TreeLinkNode *parent;
    TreeLinkNode(int x) :val(x), left(NULL), right(NULL), parent(NULL) {   
    }
};
class Solution {
public:
    TreeLinkNode* GetNextNode(TreeLinkNode* pNode){
        if(pNode == NULL)
            return NULL;
        TreeLinkNode* res = NULL;
        //有右子结点，寻找右子结点的最左子结点
        if(pNode->right){
            TreeLinkNode* tmp = pNode->right;
            while(tmp->left)
                tmp = tmp->left;
            res = tmp;
        }
        else if(pNode->parent){
            TreeLinkNode* curNode = pNode;
            TreeLinkNode* parentNode = pNode->parent;
            //结点是其父结点的右子结点，沿着指向父结点的指针一直向上遍历，直到找到一个是它父结点的左子结点的结点
            while(parentNode && curNode == parentNode->right){
                curNode = parentNode;
                parentNode = parentNode->parent;
            }
            //结点是其父结点的左子结点，那么它的下一个结点就是其父结点
            res = parentNode;
        }
        return res;
    }
};