LeetCode之二叉树的层次遍历

二叉树的层次遍历

题目：

给定一个二叉树，返回其按层次遍历的节点值。 （即逐层地，从左到右访问所有节点）。

例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
(这里排版问题 15 和 7 是20的左右节点)
返回其层次遍历结果：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/binary-tree-level-order-traversal

思路

1.用cur来存储当前层次的节点。
2.用nex存储下一次需要遍历的层次节点。
3.在遍历时把这层的值放入tmp内然后加入到res就完成了一层。之后初始化nex和tmp进行下一层。
4.直到cur大小为0则结束，返回结果即可。

代码

/**
 * 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) {
        List<List<Integer>> res = new ArrayList<>();
        if(root == null){
            return res;
        } 
        List<TreeNode> cur = new ArrayList<>();
        List<TreeNode> nex = new ArrayList<>();
        List<Integer> tmp = new ArrayList<>();
        cur.add(root);
        while(cur.size() > 0){
            for(TreeNode node : cur){
                tmp.add(node.val);
                if(node.left != null){
                    nex.add(node.left);
                }
                if(node.right != null){
                    nex.add(node.right);
                }
            }
            res.add(tmp);
            cur = nex;
            nex = new ArrayList<>();
            tmp = new ArrayList<>();
        }
        return res;
    }
}