二叉树的下一个结点

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

#include "tools.h"
using namespace std;
/*
struct TreeLinkNode {
    int val;
    struct TreeLinkNode *left;
    struct TreeLinkNode *right;
    struct TreeLinkNode *next;
    TreeLinkNode(int x) :val(x), left(NULL), right(NULL), next(NULL) {
       wn
    }
};
*/
class Solution {
public:
    TreeLinkNode* GetNext(TreeLinkNode* pNode)
    {
        if(pNode==NULL)
            return NULL;
        TreeLinkNode *pNextNode=NULL;
        if(pNode->right!=NULL)
        {
            TreeLinkNode *pRightNode=pNode->right;
            while(pRightNode->left!=NULL)
                pRightNode=pRight->left;
            pNextNode=pRightNode;
        }
        else if(pNode->next!=NULL)
        {
            TreeLinkNode *pCurrentNode=pNode;
            TreeLinkNode *pParentNode=pNode->next;
            while(pParentNode!=NULL&&pCurrentNode==pParentNode->right)
            {
                pCurrentNode=pParentNode;
                pParentNode=pCurrentNode->next;
            }
            pNextNode=pParentNode;
        }
        return pNextNode;
    }
};


你可能感兴趣的:(二叉树的下一个结点)