LeetCode 116 填充每个节点的下一个右侧节点指针【BFS】

最新推荐文章于 2024-03-17 10:45:48 发布

「已注销」

最新推荐文章于 2024-03-17 10:45:48 发布

阅读量157

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/yanzhenhuai/article/details/114547861

版权

LeetCode 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

单独把这道题提出来，不是因为这道题本身难度有多高，更多的是这道题是一道非常好的检验对树的相关知识和BFS的简单运用的题。
在这里插入图片描述

class Solution {
public:
    Node* connect(Node* root) {
        if(root == NULL)
            return root;
        queue<Node*> q; q.push(root);
        while(!q.empty()) {
            int n = q.size();
            for(int i = 1; i <= n; i++) {
                Node* node = q.front(); q.pop();
                if(i != n)
                    node -> next = q.front();
                if(node -> left != NULL) 
                    q.push(node -> left);
                if(node -> right != NULL) 
                    q.push(node -> right);
            }
        }
        return root;
    }
};