二叉树的下一个节点(非递归)

二叉树的下一个结点

http://www.nowcoder.com/questionTerminal/9023a0c988684a53960365b889ceaf5e

  • 当前节点不存在右子树,且该节点为父节点的右孩子,递归一样找父节点的中序遍历的下一个节点。
    满足这样的条件就是该节点为父节点的右孩子。
class Solution {
public:
    TreeLinkNode* GetNext(TreeLinkNode* pNode)
    {
        if (!pNode) {
            return pNode;
        }

        // 当前节点有右子树 下一个节点是右子树中序遍历的第一个节点,即下一个节点就是右孩子的最左孩子结点。
        if (pNode->right) {
            pNode = pNode->right;
            while (pNode->left) {
                pNode = pNode->left;
            }
            return pNode;
        }

        // 当前节点不存在右子树,且该节点为父节点的右孩子,递归一样找父节点的中序遍历的下一个节点。满足这样的条件就是该节点为父节点的右孩子。
        // 当前节点不存在右子树,且该节点为父节点的左孩子,直接就是其父亲节点。
        while (pNode->next) {
            TreeLinkNode *root = pNode->next;
            if (root->left == pNode) {
                return root;
            }
            pNode = pNode->next;
        }

        return nullptr;
    }
};
全部评论

相关推荐

昨天 11:21
门头沟学院 Java
总包48.5w,意想不到的价格
无情咸鱼王的秋招日记之薛定谔的Offer:R
点赞 评论 收藏
分享
牛客263158796号:我领羊一面后十天不挂也不推进 今天问hr说等前序的第一批意向发完看情况再看是否推进
点赞 评论 收藏
分享
评论
1
收藏
分享
牛客网
牛客企业服务