Leetcode之二叉树的层序遍历

    科技2022-08-02  107

    问题描述

    给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。

    示例: 二叉树:[3,9,20,null,null,15,7],

    返回其层次遍历结果:

    [ [3], [9,20], [15,7] ]

    问题解法

    先构建广度优先搜索,再设置一个指针标志这一层有多少个节点。

    /** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ class Solution { public List<List<Integer>> levelOrder(TreeNode root) { if(root==null) return new ArrayList<List<Integer>>(); Queue<TreeNode> q=new LinkedList<TreeNode>(); int end=1; TreeNode p=root; q.add(p); ArrayList<List<Integer>> list1=new ArrayList<List<Integer>>(); ArrayList<Integer> list=new ArrayList<Integer>(); while(!q.isEmpty()) { p=q.poll(); if(p.left!=null) q.add(p.left); if(p.right!=null) q.add(p.right); list.add(p.val); end--; if(end==0) { end=q.size(); list1.add(list); list=new ArrayList<Integer>() ; } } return list1; } }

    运行结果

    Processed: 0.008, SQL: 8