前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >二叉树同层处理

二叉树同层处理

作者头像
摸鱼的G
发布2023-02-22 11:56:13
1620
发布2023-02-22 11:56:13
举报
文章被收录于专栏:火属性小虫

二叉树同层处理

个人整理,非最优解


二叉树算法中,如果我们需要如:获取每一层二叉树的元素总和时,这样对二叉树每一层进行分开计算,可以使用队列加两层while循环:

代码语言:javascript
复制
public void aSum(TreeNode root) {
        int floor=0;
        Deque<TreeNode> queue = new ArrayDeque<>(); //队列
        queue.offer(root); //初始根节点压入队列
        while(!queue.isEmpty()){ //第一层while,知道二叉树被遍历完全
            floor++; //表示现在正在计算的层数
            int size=queue.size(); //通过获取此时的size来表示该层有多少元素
            int sum=0;
            while(size!=0){ //第二层while,通过size来遍历同层的元素
                size--; //每遍历一个同层元素减1
                TreeNode ls=queue.pop(); //将元素出队列
                sum+=ls.val;
                if(ls.left!=null){queue.offer(ls.left);} //左分支加入队列
                if(ls.right!=null){queue.offer(ls.right);} //右分支加入队列
            }

        }

    }

如上面的算法,我们先将根节点压入队列,之后通过一层while(和之后的左右分支入队)来实现二叉树的遍历。之后通过遍历floor来记录当前计算的层数,并通过size表示这个层有多少元素。

我们通过size为基准,再次使用while,此时遍历的为当前floor的所有元素,并将左右分支入队。

这个方法主要使用了记录size,通过每次遍历完一层,将元素全部出队,并在下一层循环开始前记录此时的size,来区分每一层有几个元素,以此实现方法。

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 二叉树同层处理
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档