leetcode 日经贴，Cpp code -Populating Next Right Pointers in Each Node II

原创于 2015-05-03 00:07:03 发布 · 303 阅读

0 ·

CC 4.0 BY-SA版权

算法同时被 3 个专栏收录

183 篇文章

订阅专栏

leetcode

173 篇文章

订阅专栏

Cpp

152 篇文章

订阅专栏

本文介绍了一种填充二叉树中每个节点的next指针的方法，使得每一个节点的next指针指向其下一个右侧的节点。通过递归方式实现了这一过程，确保了树的每一层节点都被正确连接。

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

Populating Next Right Pointers in Each Node II

/**
 * Definition for binary tree with next pointer.
 * struct TreeLinkNode {
 *  int val;
 *  TreeLinkNode *left, *right, *next;
 *  TreeLinkNode(int x) : val(x), left(NULL), right(NULL), next(NULL) {}
 * };
 */
class Solution {
public:
    void update(TreeLinkNode *cur, TreeLinkNode *p) {
        p = p->next;
        while (p) {
            if (p->left) {
                cur->next = p->left;
                cur = cur->next;
            }
            if (p->right) {
                cur->next = p->right;
                cur = cur->next;
            }
            p = p->next;
        }
        cur->next = NULL;
    }
    void con(TreeLinkNode *r) {
        if (!r) return;
        TreeLinkNode *c = r->right? r->right:r->left;
        if (r->left && r->right) {
            r->left->next = r->right;
        }
        if (c && !c->next) {
            update(c, r);
        }
        con(r->left);
        con(r->right);
    }
    void connect(TreeLinkNode *root) {
        if (!root) return;
        root->next = NULL;
        con(root);
    }
};