码迷,mamicode.com
首页 > 其他好文 > 详细

LeetCode 树 116. 填充每个节点的下一个右侧节点指针(层序遍历 分层)

时间:2020-07-13 18:42:31      阅读:79      评论:0      收藏:0      [点我收藏+]

标签:死循环   height   idt   com   color   array   node   层序遍历   计数   

技术图片

 

 技术图片

 

 题目的要求,大白话说就是:把指针指向同层的右侧节点

提到同层,自然就要想到层序遍历,自然是队列实现

 

 

问题是需要分层

所以采用之前题目用的那种计数的方式

class Solution {
    public Node connect(Node root) {
        if(root==null)
        {return root;}
        
        Queue<Node> queue=new ArrayDeque<>();
        Node fake=root;
        queue.add(root);
        while(queue.size()!=0)
        {
            int count=queue.size();
            for(int n=0;n<count;n++)
            {
                fake=queue.poll();
                if(n<count-1)
                {
                    fake.next=queue.peek();
                }
                if(fake.left!=null)
                {
                    queue.add(fake.left);
                }
                if(fake.right!=null)
                {
                    queue.add(fake.right);
                }
            }
        }
        return root;
    }
}

几个碰到的问题:

1.前面要加对于空树的判断,要不会搞出null.next这种错误

2.循环跳出的条件是什么?是queue.size()!=0,即所有元素已经进了队列并且弹出了。我一开始写成了fake!=null这个条件。仔细走一下这个顺序,发现fake会一直都是最后一个节点,跑成了死循环。但其实加个判断也能解决,但其实判断的条件就是队列排空。

3.这个地方

if(n<count-1)
                {
                    fake.next=queue.peek();
                }

首先是,好在有一个peek的方法,使我们不用搞front Node和back Node来把先后排出队列的节点连起来

其次是为什么要加if(n<count-1)这个计数的条件。因为最后一个元素的peek会peek到下一层去。。。

 

由此发现,这种循环中,判断循环条件是什么任然是一个很棘手的问题。

 

LeetCode 树 116. 填充每个节点的下一个右侧节点指针(层序遍历 分层)

标签:死循环   height   idt   com   color   array   node   层序遍历   计数   

原文地址:https://www.cnblogs.com/take-it-easy/p/13294628.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!