leetcode 102.二叉树的层序遍历

难度:中等
频率:150

题目:
给你一个二叉树,请你返回其层序遍历得到的节点值(逐层,从左到右访问所有节点)
在这里插入图片描述

做这道题 之前先回忆一下数据结构里的 BFS(Breath First Search,广度优先搜索)和DFS(Depth First Search)如何实现遍历。

  • DFS[递归方法]
void dfs(TreeNode root)
{
	if(root!=null)
	dfs(root.left);
	dfs(root.right)}
  • BFS[使用队列数据结构]
void bfs(TreeNode root)
{
  Queue<TreeNode> queue=new ArrayDeque<>();
  queue.add(root);
  while(!queue.isEmpty()){
		TreeNode node = queue.poll();
		if(node.left!=null)
		{
			queue.add(node.left);
		}
		if(node.right!=null)
		{
			queue.add(node.right);
		}
  }
}

总结:BFS用来解决层级遍历问题,而DFS用来解决最短路径问题。

解题方法:BFS变形
解题思路:
1.在常规的BFS遍历中【队列】,加入一个参数,记录这一层的节点数量。
比如一开始root进入的时候,记录数量1。判断一左一右为一次。计次1次遍历,即遍历一棵小树。
然后进入2个,这个时候就会记录2。然后倒计时遍历2次,即遍历两个小树,或者理解为2个root。
2.每一次倒计时用一个arraylist,每次队列出去的数,都放在arraylist里。
3.最后再用一个大的arraylist,把之前每一次的arraylist装起来。

/**
 * 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;
 *     }
 * }
 */
class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> res=new ArrayList<>();
        Queue<TreeNode>queue=new ArrayDeque();
        if(root!=null)
        queue.add(root);
        while(!queue.isEmpty()){
            int n=queue.size();
            List<Integer> arr = new ArrayList();
            for(int i=0;i<n;i++)
            {
                TreeNode node =queue.poll(); 
                arr.add(node.val);
                if(node.left!=null)
                {
                    queue.add(node.left);
                }
                if(node.right!=null)
                {
                    queue.add(node.right);
                }
            }
            res.add(arr);
        }  return res;
    }
  
}

PS:易错点

  • 双向队列里面装的是树节点,不是普通类型。
  • 一开始的根节点需要判断是否为空。
  • list添加的都是值,不是树节点。
  • 队列的里节点的数量 是 queue.size()。
  • list里套list。二位数组。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值