面试必考真题-算法篇 牛客网
树 DFS
题目描述
给定一个二叉树,返回该二叉树层序遍历的结果,(从左到右,一层一层地遍历)
题目分析
层次遍历完事。
注意每层遍历完之后不要清空ArrayList,而是要重新new 一个 ArrayList,如果清空了,返回值的结果也会清空。
下面是Java代码
import java.util.*;
/*
* public class TreeNode {
* int val = 0;
* TreeNode left = null;
* TreeNode right = null;
* }
*/
public class Solution {
/**
*
* @param root TreeNode类
* @return int整型ArrayList<ArrayList<>>
*/
public ArrayList<ArrayList<Integer>> levelOrder (TreeNode root) {
// write code here
ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();
if(root == null){
return list;
}
ArrayList<Integer> tmp = new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
TreeNode cur = null;
int count = 0 ;
int nextCount = 1;
while(!queue.isEmpty()){
cur = queue.poll();
++count;
tmp.add(cur.val);
if(cur.left!=null){
queue.offer(cur.left);
}
if(cur.right!=null){
queue.offer(cur.right);
}
if(count == nextCount){
count = 0;
nextCount = queue.size();
list.add(tmp);
//注意要new一个ArrayList,不能清空原有的ArrayList。
tmp = new ArrayList<>();
}
}
return list;
}
}