遍历 Populating Next Right Pointers in Each Node II

最新推荐文章于 2024-08-25 16:11:27 发布

qwurey

最新推荐文章于 2024-08-25 16:11:27 发布

阅读量392

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/yeruby/article/details/45744831

leetcode 专栏收录该内容

234 篇文章

订阅专栏

本文介绍了一种用于解决树结构中节点连接问题的高效算法，通过逐步迭代实现上下层节点间的连接，确保了数据结构的正确性和效率。

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

思路：

跟上一题思路一样：只要上一层next指针串起来了该层所有对象，那么下一层就可以串起来。

/**
 * 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 {
private:
    TreeLinkNode* findNextNodeNextLevel(TreeLinkNode* &cur, TreeLinkNode* curNextLevel) {
        if(cur->left == curNextLevel && cur->right != NULL) return cur->right;
        else {
            while(cur->next != NULL) {
                cur = cur->next;
                if(cur->left != NULL && cur->left != curNextLevel) return cur->left;
                if(cur->right != NULL && cur->right != curNextLevel) return cur->right;
            }
        }
        return NULL;
    }
    TreeLinkNode* findStartNodeNextLevel(TreeLinkNode* node) {
        if(node == NULL) return NULL;
        do {
            if(node->left != NULL) return node->left;
            if(node->right != NULL) return node->right;
            node = node->next;
        }while(node != NULL);
        return NULL;
    }
public:
    void connect(TreeLinkNode *root) {
        if(root == NULL) return;
        TreeLinkNode *start, *cur, *next;
        while(root != NULL) {
            start = findStartNodeNextLevel(root);
            cur = start;
            next = findNextNodeNextLevel(root, start);
            while(next != NULL) {
                cur->next = next;
                cur = next;
                next = findNextNodeNextLevel(root, cur);
            }
            root = start;
        }
    }
};