题解 | #求二叉树的层序遍历#

求二叉树的层序遍历

https://www.nowcoder.com/practice/04a5560e43e24e9db4595865dc9c63a3

<?php

/*class TreeNode{
    var $val;
    var $left = NULL;
    var $right = NULL;
    function __construct($val){
        $this->val = $val;
    }
}*/

/**
 * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
 *
 * 
 * @param root TreeNode类 
 * @return int整型二维数组
 */
function levelOrder( $root )
{
    if($root == null){
        return [];
    }
    $res = []; // 返回结果
    $list = []; // 队列
    $list[] = $root;
    $levelNum = [];
    $levelNum[0] = 1; // 第一层是1个
    $currLevel = 0;
    $currLevelPop = 0;
    while(!empty($list)){
        $pop = array_shift($list);
        $currLevelPop++;
        if($pop->left != null){
            $list[] = $pop->left;
            $levelNum[$currLevel+1]++;
        }
        if($pop->right != null){
            $list[] = $pop->right;
            $levelNum[$currLevel+1]++;
        }
        if(empty($res[$currLevel])){
            $res[$currLevel] = [];
        }
        $res[$currLevel][] = $pop->val;
        if($currLevelPop == $levelNum[$currLevel]){
            // 本层已经pop完了
            $currLevel++;
            $currLevelPop = 0;
        }
    }
    return $res;
}

层序遍历:每一层从左到右的遍历,=BFS广度优先搜索

方法:队列法。每次从队列左边pop一个元素,就把pop的这个元素的子节点从队列右边放入队列。本题的难点在于需要分层输出,如果不需要分层从队列过一遍即可。

全部评论

相关推荐

05-12 11:09
已编辑
门头沟学院 后端
已注销:没必要放这么多专业技能的描述。这些应该是默认已会的,写这么多行感觉在凑内容。项目这块感觉再包装包装吧,换个名字,虽然大家的项目基本都是网上套壳的,但是你这也太明显了。放一个业务项目,再放一个技术项目。技术项目,例如中间件的一些扩展和尝试。
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务