每天1题算法题(2)-二叉树的层序遍历

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

 

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

3
/ \
9 20
/ \
15 7
返回其层次遍历结果:

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

解答

1.使用广度优先算法,加上level标记当前层

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        // queue先进先出,每次放入到尾部,每次在首部移出
        List<List<Integer>> response = new ArrayList<List<Integer>>();
        if(root == null) {
            return response;
        }
        Queue<TreeNode> queue = new LinkedList();
        queue.offer(root);
        while(!queue.isEmpty()) {
            int currentLevelSize = queue.size();
            List<Integer> levelTreeNodes = new ArrayList();
            for(int i=0; i<currentLevelSize; i++) {
                TreeNode treeNode = queue.poll();
                if(treeNode.left != null) {
                    queue.offer(treeNode.left);
                }
                if(treeNode.right != null) {
                    queue.offer(treeNode.right);
                }

                levelTreeNodes.add(treeNode.val);
            }
            response.add(levelTreeNodes);
        }
        return response;
    }
}

 

posted @ 2020-09-17 19:44  GodTelMe  阅读(170)  评论(0编辑  收藏  举报