leetcode populating-next-right-pointers-in-each-node 链表中同一层的结点连接起来

原创于 2019-04-02 03:47:37 发布 · 134 阅读

0 ·

CC 4.0 BY-SA版权

刷题同时被 3 个专栏收录

36 篇文章

订阅专栏

leetcode

31 篇文章

订阅专栏

链表

7 篇文章

订阅专栏

本文介绍了一种算法，用于将完全二叉树每一层的节点通过新增的next指针连接起来，实现了广度优先搜索（BFS）。通过使用队列结构，此方法能有效地遍历树的所有层级并将相邻节点链接。

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

简单点就是，给定完全二叉树，将它每一层的顶点串起来

Given a binary tree

    struct TreeLinkNode {
      TreeLinkNode *left;
      TreeLinkNode *right;
      TreeLinkNode *next;
    }

For example
Given the following perfect binary tree,

For example,
Given the following perfect binary tree,
         1
       /  \
      2    3
     / \  / \
    4  5  6  7

After calling your function, the tree should look like:
         1 -> NULL
       /  \
      2 -> 3 -> NULL
     / \  / \
    4->5->6->7 -> NULL

用BFS做

/**
 * Definition for binary tree with next pointer.
 * public class TreeLinkNode {
 *     int val;
 *     TreeLinkNode left, right, next;
 *     TreeLinkNode(int x) { val = x; }
 * }
 */
import java.util.*;
public class Solution {
    public void connect(TreeLinkNode root) {
        if(root==null) return ;
        LinkedList<TreeLinkNode> que = new LinkedList<>();
        que.add(root);
        while(!que.isEmpty()){
            int size = que.size();
            for(int i=0;i<size;i++){
                TreeLinkNode cur = que.poll();//获取并删除
                if(i==size-1){
                    cur.next=null;//该层最后一个结点
                }
                else{
                    cur.next = que.peek();//获取que中的第一个元素但不删除
                }
                if(cur.left!=null){
                    que.add(cur.left);
                }
                if(cur.right!=null){
                    que.add(cur.right);
                }
            }
        }
    }
}