【牛客题解】 ——二叉树的下一个节点

二叉树的下一个节点

【牛客题解】 ——二叉树的下一个节点_第1张图片
题中给出的指向父亲节点的指针,用next表示,有点会让人误解。

思路:我们以以下这棵二叉树为例,中序遍历为{D,B,H,E,I,A,F,C,G}
【牛客题解】 ——二叉树的下一个节点_第2张图片
把很容易可以发现,如果要找一个节点的下一结点,比如说是B节点的下一结点,我们首先看B的右树是不是为空,如果不为空,一直沿着指向左子结点的指针找到的叶子节点即为下一个节点,在这里B的下一个节点是H;

无右子树,且结点是该结点父结点的左子树,则下一结点是该结点的父结点,例如 H,下一结点是 E

无右子树,且结点是该结点父结点的右子树,则我们一直沿着父结点追朔,直到找到某个结点是其父结点的左子树,如果存在这样的结点,那么这个结点的父结点就是我们要找的下一结点。例如 I,下一结点是 A;例如 G,并没有符合情况的结点,所以 G 没有下一结点

/*
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;
        if(pNode->right!=NULL)  //右树不为空,右树的最左孩子就是他的下一结点
        {
            pNode=pNode->right;
            if(pNode->left!=NULL)
                pNode=pNode->left;
            return pNode;
        }
        while(pNode->next)  //右树为空
        {
            TreeLinkNode* parent=pNode->next;
            if(parent->left==pNode)
                return parent;
            pNode=pNode->next;
        }
        return NULL;
    }
};

你可能感兴趣的:(C++,数据结构,二叉树,数据结构)