之字遍历二叉树并填充next指针

最新推荐文章于 2024-12-15 16:32:07 发布

JAVA程序员——

最新推荐文章于 2024-12-15 16:32:07 发布

阅读量793

点赞数

CC 4.0 BY-SA版权

分类专栏：面试总结文章标签： java 数据结构

本文链接：https://blog.youkuaiyun.com/Juwenzhe_HEBUT/article/details/122812803

面试总结专栏收录该内容

33 篇文章

订阅专栏

该博客主要讨论了一种特定的二叉树结构——完美二叉树，并提出了一种算法来填充其节点的next指针，使其指向右侧相邻节点。通过使用层次遍历（之字形遍历）的方法，遍历树的过程中更新next指针，确保在没有右侧节点的情况下将其设置为NULL。这种方法优化了树结构的数据连接，提高了数据操作效率。

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

//蚂蚁一面评测题目: 给定一个完美二叉树，其所有叶子节点都在同一层，每个父节点都有两个子节点，
//填充它的每个 next 指针，让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点，则将 next 指针设置为 NULL。
//初始状态下，所有 next 指针都被设置为 NULL。

public class Test {
  
  public static void main(String[] args) {
    // pointToNext(root);
  }
  
  public static void pointToNext(TreeNode root) {
    if(root == null) return;
    // 之字遍历用队列
    List<TreeNode> queue = new ArrayList<>();
    queue.add(root);
    while(queue.size() != 0){
      int i=0;
      // 遍历的同时，搞定next指针
      while(i < queue.size()-1){
        queue.get(i).next = queue.get(i+1);
        i++;
      }
      
      for(int j=0; j <= i; j++){
        TreeNode head = queue.get(0);
        queue.add(head.left);
        queue.add(head.right);
        head.remove(0);
      }
    }
  }
}