题目描述:
思路:
层序遍历的一般写法:通过一个while循环控制从上向下一层层遍历,for循环控制每一层从左向右遍历:
代码实现:
/**
* 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;
* }
* }
*/
// 层序遍历的一般写法:通过一个while循环控制从上向下一层层遍历,for循环控制每一层从左向右遍历
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
// 根据题目结果创建一个二维数组来存取最后的结果
List<List<Integer>> res = new LinkedList<>();
if (root == null){
return res;
}
// 再创建一个队列q来进行存取
Queue<TreeNode> q = new LinkedList<>();
q.offer(root);
// while循环控制从上往下一层层遍历
while(!q.isEmpty()){
int sz = q.size();
// 创建一个一维数组来记录这一层的节点值
List<Integer> level = new LinkedList<>();
// for循环控制每一层从左到右遍历
for(int i = 0; i < sz; i++){
// 从左往右每次都弹出一个元素
TreeNode cur = q.poll();
level.add(cur.val);
if (cur.left != null)
q.offer(cur.left);
if (cur.right != null)
q.offer(cur.right);
}
// 将每一层得到的一维数组存入二维数组当中
res.add(level);
}
return res;
}
}
注意点:
- 创建一个二维数组的形式:
List<List<Integer>> res = new LinkedList<>();
- 创造一个一维数组的形式:
Queue<TreeNode> q = new LinkedList<>();
- Java队列Queue中offer/add函数,poll/remove函数,peek/element 函数的区别:实际使用中还是需要注意一下!!!