leetcode 102 - 二叉树的层次遍历

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

例如:
给定二叉树: [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回其层次遍历结果:

[
  [3],
  [9,20],
  [15,7]
]
public List<List<Integer>> levelOrder(TreeNode root) {
    List<List<Integer>> result = new ArrayList<>();
    List<Integer> rowResult = new ArrayList<>();
    if(root == null){
       return result;
    } 
    int next = 0; 
    int num = 1;
    Queue<TreeNode> queue = new LinkedList<>();
    queue.add(root);
    while(!queue.isEmpty()){
        TreeNode node = queue.poll();
        rowResult.add(node.val);
        num--;
        if(node.left != null){
            queue.add(node.left);
        }
        if(node.right != null){
            queue.add(node.right);
        }
        if(num == 0){
            num = next;
            next = 0;
            result.add(rowResult);
            rowResult = new ArrayList<>();
        }
    }
    return result;
}

猜你喜欢

转载自blog.csdn.net/qq_43322057/article/details/84402730
今日推荐