专业的JAVA编程教程与资源

网站首页 > java教程 正文

java 二叉树的层次遍历之队列的使用

temp10 2024-09-08 09:40:05 java教程 9 ℃ 0 评论

二叉树的层次遍历可以使用队列来实现。以下是使用队列进行二叉树的层次遍历的基本步骤:

  • 创建一个空队列。
  • 将根节点入队。
  • 开始循环,直到队列为空:

1、弹出队列中的节点,将其值加入结果列表。

java 二叉树的层次遍历之队列的使用

2、如果有左子节点,则将左子节点入队。

3、如果有右子节点,则将右子节点入队。

  • 返回结果列表。

以下是java 代码实现

/**
 * 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) {
        if(root==null){
            return Collections.emptyList();
        }
        List<List<Integer>> result=new ArrayList<>();
        Deque<TreeNode> queue=new ArrayDeque<>();
        queue.offer(root);
        while(!queue.isEmpty()){
            //当前队列的元素个数,也就是该层节点元素的个数
            int size=queue.size();
            //集合存储当前层的元素
            List<Integer> list=new ArrayList<>();
            for(int i=0;i<size;i++){
                //出队操作
                TreeNode temp=queue.poll();
                //将出队列的元素添加到当前层的集合当中
                list.add(temp.val);
                //如果出队的元素左节点不为空,则入队列
                if(temp.left!=null){
                    queue.offer(temp.left);
                }
                //如果出队的元素右节点不为空,则入队列
                if(temp.right!=null){
                    queue.offer(temp.right);
                }
            }
            result.add(list);
        }
        return result;
    }
}

Tags:

本文暂时没有评论,来添加一个吧(●'◡'●)

欢迎 发表评论:

最近发表
标签列表