题解 | #牛群左侧视图# java

牛群左侧视图

https://www.nowcoder.com/practice/1eba2dd947484c078e6359ccd90aa7b1

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 *   public TreeNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     *
     * @param root TreeNode类
     * @return int整型一维数组
     */
    public int[] leftSideView (TreeNode root) {
        // write code here
        List<Integer> ans = new ArrayList<>();
        if (root == null)
            return new int[0];

        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            int n = queue.size();
            for (int i = 0; i < n; i++) {
                TreeNode node = queue.poll();
                if (i == 0)
                    ans.add(node.val);

                if (node.left != null)
                    queue.offer(node.left);
                if (node.right != null)
                    queue.offer(node.right);
            }
        }

        int[] result = new int[ans.size()];
        for (int i = 0; i < ans.size(); i++) {
            result[i] = ans.get(i);
        }
        return result;
    }
}

该代码使用的编程语言是Java。

该题考察的是二叉树的层次遍历,并返回每一层的最左侧节点的值。

leftSideView方法中,首先创建一个vector<int>类型的变量ans来存储结果。

然后判断根节点是否为空,如果为空,则直接返回一个空数组。

使用一个队列qu来进行层次遍历。将根节点入队。

然后开始循环,直到队列为空。在每次循环中,先获取当前层级的节点数量n

然后使用一个内层循环,循环遍历当前层级的节点。将节点出队。

如果是当前层级的第一个节点(即i==0),则将节点的值加入到ans中。

如果节点的左子节点不为空,则将左子节点入队;如果节点的右子节点不为空,则将右子节点入队。

内层循环结束后,将ans转换为数组并返回作为结果。

全部评论

相关推荐

Hyh_111:像这种hr就不用管了,基本没啥实力,换一个吧
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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