LeetCode[Tree]: Populating Next Right Pointers in Each Node II

最新推荐文章于 2023-03-11 16:53:38 发布

原创最新推荐文章于 2023-03-11 16:53:38 发布 · 847 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #迭代 #tree #二叉树 #算法

LeetCode 同时被 2 个专栏收录

74 篇文章

订阅专栏

LeetCode

72 篇文章

订阅专栏

针对LeetCode上的二叉树题目，介绍如何在不使用额外空间的情况下填充每个节点的next指针，使其指向同一层级的下一个节点。适用于任意二叉树结构。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Follow up for problem "Populating Next Right Pointers in Each Node".
What if the given tree could be any binary tree? Would your previous solution still work?
Note:
You may only use constant extra space.
For example,
Given the following binary tree,
1
/ \
2 3
/ \ \
4 5 7
After calling your function, the tree should look like:
1 -> NULL
/ \
2 -> 3 -> NULL
/ \ \
4-> 5 -> 7 -> NULL

这个题目是LeetCode[Tree]: Populating Next Right Pointers in Each Node的后续，这里的迭代思路与之相同，只有一个地方的差别：由于树有可能是任意的二叉树，因此当获取下一个节点的时，需要用一个循环进行查找。

我的C++的代码实现如下：

class Solution {
public:
    void connect(TreeLinkNode *root) {
        for (TreeLinkNode *levelFirstNode = root; levelFirstNode != nullptr; levelFirstNode = getNext(levelFirstNode)) {
            for (TreeLinkNode *curNode = levelFirstNode; curNode != nullptr; curNode = curNode->next) {
                if (curNode->left) curNode->left->next = curNode->right ? curNode->right : getNext(curNode->next);
                if (curNode->right) curNode->right->next = getNext(curNode->next);
            }
        }
    }

private:
    TreeLinkNode *getNext(TreeLinkNode *node) {
        while (node) {
            if (node->left) return node->left;
            if (node->right) return node->right;
            node = node->next;
        }

        return nullptr;
    }
};