57.二叉树的下一个结点

题目

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

思路

• 中序遍历遍历的顺序是左中右。
• 如果一个结点的右孩子不为空，那么有两种情况。
  • 右孩子没有左孩子，那么该右孩子即为下一个结点。
  • 如果右孩子还有左孩子，那么一路找左孩子，找到最后一个。
• 如果右孩子为空，检查自己的父结点是不是某一个结点的左孩子。
  • 如果是的话，那么父结点的父结点就是下一个结点。
  • 如果不是的话，一路找到第一个为左孩子的祖先结点。该祖先结点的父结点便为下一个结点。
  • 注意根节点的父结点为空。

代码

/*
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)
    {
        TreeLinkNode* resNode = nullptr;
        
        if ( pNode == nullptr )    // 空结点
            return resNode;
        
        if (  pNode->right ) {
            resNode =  findFarLeftNode( pNode->right );
        }
        else if ( !pNode->right && pNode->next ) {
            
            if ( pNode == pNode->next->left ) {
                resNode = pNode->next;
            }
            else if ( pNode == pNode->next->right ) { // 找到一个祖先节点是左孩子
                while ( pNode->next ) {
                    pNode = pNode->next;
                    
                    if ( pNode->next && pNode == pNode->next->left )
                        break;
                }
                
                resNode = pNode->next;
            }
        }
       
        return resNode;
    }
    
    TreeLinkNode* findFarLeftNode( TreeLinkNode* node ) {
        while ( node && node->left )
            node = node->left;
        
        return node;
    }
};