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

题目
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
思路
根据中序遍历的特点,要找到一个节点的下一个节点无非就是三种情况:1、有右子树,这时只需要把其右孩子作为下一个遍历的(并不是要找的)节点,然后沿着该节点的左子树(如果有的话)出发,直到遇到叶子节点,那么该叶子节点就是其下一个要找的节点;2、没有右子树,则判断该节点是否是其父节点的左孩子,如果是则其下一个要找的节点是其父节点;3、如果不是其父节点的左孩子,则把其父节点作为下一个遍历的节点,向上回溯,直到找到父节点没有父节点并且父节点是父节点的父节点的左孩子为止。综合这三种情况就可以找到二叉树中任意一个节点的下一个节点。

struct BinaryTreeNode
{
    int m_nValue;
    BinaryTreeNode*m_nLeft;
    BinaryTreeNode*m_nRight;
    BinaryTreeNode*m_nParent;
};
BinaryTreeNode*FindNext(BinaryTreeNode*root,BinaryTreeNode*target)
{
    if(root==NULL||target==NULL)
        return NULL;

    if(target->m_nRight!=NULL)
        target=target->m_nRight;
        while(target->m_nLeft!=NULL)
            target=target->m_nLeft;
        return target;

    if(target->m_nRight==NULL&&target->m_nParent!=NULL)
    {
        if(target->m_nParent->m_nLeft=target)
            return target->m_nParent;

        else 
        {
            while(target->m_nParent!=target->m_nParent->m_nParent->m_nLeft)
                target=target->m_nParent;
            return target->m_nParent->m_nParent;
        }
    }
    return NULL;

}

猜你喜欢

转载自blog.csdn.net/gary_god/article/details/79620443