Leetcode-102. 二叉树的层序遍历

今天的情人节和树过了......

题目:

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

示例 1:

Leetcode-102. 二叉树的层序遍历_第1张图片

输入:root = [3,9,20,null,null,15,7]
输出:[[3],[9,20],[15,7]]

示例 2:

输入:root = [1]
输出:[[1]]

示例 3:

输入:root = []
输出:[]

提示:

  • 树中节点数目在范围 [0, 2000] 内
  • -1000 <= Node.val <= 1000

 层次遍历二叉树,不得不想到BFS(广度优先搜索),使用队列(FIFO的特性),实现分层遍历。从根结点开始,先存放入队列当中,如果这一层结束,就把值放入lever当中,然后再判断该结点下左子树和右子树是否存在,存在就入队。(总之,就是一层全部入队,再一层全部出队)

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public List> levelOrder(TreeNode root) {
            List> result = new ArrayList<>();

            Queue queue = new LinkedList<>();
            if(root !=null){
                queue.add(root);
            }
            while(!queue.isEmpty()){
                int n = queue.size();
                List lever = new ArrayList<>();
                for(int i = 0; i < n ; i++){
                    TreeNode node = queue.poll();
                    lever.add(node.val);
                    if(node.left!=null){
                        queue.add(node.left);
                    }
                    if(node.right!=null){
                        queue.add(node.right);
                    }

                }
                result.add(lever);
            }
            return result;
    }
}

 

你可能感兴趣的:(leetcode,算法,职场和发展,java)