leetcode 117.Populating Next Right Pointers in Each Node II

蓝羽飞鸟

已于 2022-05-13 11:59:46 修改

阅读量98

点赞数

分类专栏： leetcode 文章标签：算法 leetcode

于 2019-07-26 22:17:51 首次发布

本文链接：https://blog.csdn.net/level_code/article/details/97425951

版权

leetcode 专栏收录该内容

490 篇文章 4 订阅

订阅专栏

在这里插入图片描述
给一个二叉树，每个node有个next指针，让每层node的next指针指向它右边的node，最右边的next指向null。

思路：
BFS
每层的node装入queue, queue中的node指向下一node，每层的最后一个node.next保持null。

    public Node connect(Node root) {
        if(root == null) return root;
        
        Queue<Node> queue = new LinkedList<>();
        
        queue.offer(root);
        
        while(queue.size() > 0) {
            int size = queue.size();
            
            for(int i = 0; i < size; i ++) {
                Node cur = queue.poll();
                if(i < size - 1) cur.next = queue.peek();
                
                if(cur.left != null) queue.offer(cur.left);
                if(cur.right != null) queue.offer(cur.right);
            }
        }
        
        return root;
    }