题解 | #二叉树的下一个结点#

二叉树的下一个结点

https://www.nowcoder.com/practice/9023a0c988684a53960365b889ceaf5e

基本思路:

  • 如果有右子树,找右子树中的最左节点
  • 如果没有右子树,向上遍历直到根节点或者为某个树的左子树,中序遍历下一个节点即为该左子树的根节点
class Solution {
public:
    TreeLinkNode* GetNext(TreeLinkNode* pNode) {
        if (!pNode) return nullptr;
        TreeLinkNode* res;
        if (pNode->right) {
            res = pNode->right;
            while (res->left) res = res->left;
            return res;
        }
        res = pNode;
        while (res->next && res == res->next->right) res = res->next;
        return res->next;
    }
};

全部评论

相关推荐

评论
点赞
收藏
分享
牛客网
牛客企业服务