8题 剑指offer 二叉树的下一个结点

题目描述

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

思路1：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:
    TreeLinkNode* GetNext(TreeLinkNode* pNode)
    {
        if(pNode==NULL)
          return NULL;
        TreeLinkNode*pnext=NULL;
        if(pNode->right)//如果结点存在右子树，那么其右子树的最左结点就是他的下一个结点
        {
            TreeLinkNode*prNode=pNode->right;
            while(prNode->left)
                prNode=prNode->left;
            pnext=prNode;
        }//如果结点不存在右子树，分为两种情况
        else 
           if(pNode->next)
             {   TreeLinkNode*pcurrent=pNode;
                TreeLinkNode*pparent=pNode->next;
                //结点不是父节点的左结点，则向上遍历，直到某一结点是其父节点的左结点，其下一节点就是那个父节点
                 while(pparent&&pcurrent==pparent->right)
                     {
                         pcurrent=pparent;
                         pparent=pparent->next;
                      }
                 pnext=pparent;//结点是父节点的左结点，则其下一节点就是父节点
             } 
         return pnext;
    }
};

思路2：

1.二叉树为空，则返回空；

2.节点右孩子存在，则设置一个指针从该节点的右孩子出发，一直沿着指向左子结点的指针找到的叶子节点即为下一个节点；

3.节点不是根节点。如果该节点是其父节点的左孩子，则返回父节点；否则继续向上遍历其父节点的父节点，重复之前的判断，返回结果。

代码：

class Solution {
public:
    TreeLinkNode* GetNext(TreeLinkNode* pNode)
    {
        if(pNode==NULL)
            return NULL;
        if(pNode->right)
        {
            pNode=pNode->right;
            while(pNode->left!=NULL)
                pNode=pNode->left;
            return pNode;
        }  
        while(pNode->next)
        {
            TreeLinkNode *proot=pNode->next;
            if(proot->left==pNode)
                return proot;
            pNode=pNode->next;
        }
        return NULL;
    }
};