剑指offer - 57二叉树的下一个结点

题目描述

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

思路如下：

定义的结构体中next指向的是父结点

中序遍历是：左子树 - 根 - 右子树

因此某个节点的下一个节点有下面几种情况：

1. 当前节点的右子树不空，则下一个节点是其右子树中序遍历的第一个节点 （沿着右子树一直向左找第一个叶子节点）

2. 如果右子树为空，又分两种情况

         1> 如果此节点是其父节点的左孩子，则其父节点就是下一个节点

         2> 如果此节点是其父节点的右孩子，则其已经是最后一个节点了， 需要循环向上找其父节点的下一个节点

代码如下:
 

/*
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:
    //中序遍历： 左子树 - 根 - 右子树
    //因此一个节点的下一个节点有下面几种情况：
    //1. 此节点的右子树不空 则下一个节点是右子树中序遍历的第一个节点（也就是沿着右子树的左子树一直向左）
    //2. 如果右子树为空
    //      如果此节点是其父节点的左孩子 则其父节点就是下一个节点
    //      如果此节点是其父节点的右孩子 则此节点已经是其父子树的最后一个节点了 需要找到其父节点的下一个节点
    //                                因此循环判断 找到第一个子节点是父节点的左子树的情况 此时 父节点即为下一个节点
    TreeLinkNode* get_inorder_nextnode(TreeLinkNode* pNode){
        if(NULL == pNode){
            return NULL;
        }
        if(pNode->right){   //右子树不空的情况
            TreeLinkNode* node = pNode->right;  
            while(node->left){     //找到右子树中序遍历的第一个节点 一直向左 直到第一个叶子节点
                node = node->left;
            }
            return node;
        }
        else{              //右子树为空的情况
            TreeLinkNode* subnode = pNode;
            TreeLinkNode* parnode = pNode->next;
            while(parnode){
                if(subnode == parnode->left){  //如果当前节点是其父节点的左子树 则父节点就是下一个节点
                    return parnode;
                }
                else{                          //如果当前节点是其父节点的右子树 则循环判断父节点的父节点
                    subnode = parnode;
                    parnode = parnode->next;
                }
            }
        }
        return NULL;
    }

    TreeLinkNode* GetNext(TreeLinkNode* pNode)
    {
        return get_inorder_nextnode(pNode);
    }

};