算法-BFS广度优先搜索

桑英豪

已于 2022-03-23 16:48:53 修改

阅读量345

点赞数

分类专栏：算法

于 2022-03-23 15:58:47 首次发布

本文链接：https://blog.csdn.net/m0_62902168/article/details/123687995

版权

二叉树层次遍历广度优先搜索队列数据结构

关键词由CSDN通过智能技术生成

算法专栏收录该内容

25 篇文章 6 订阅

订阅专栏

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        //新建一个二维数组
        List<List<Integer>> result =new ArrayList<>();
        //如果根节点为空，直接返回空了
        if(root==null){
            return result;
        }
        //新建一个队列
        Queue<TreeNode> q = new LinkedList<>();
        //把根节点添加进队列
        q.add(root);
        //一个循环是一层
        while (q.size()>0){
            //把当前层的数据量记录下来
            int size=q.size();
            //新建一个一维数组，用来记录每一层
            ArrayList<Integer> list = new ArrayList<>();
            //遍历当前层的所以数据，把每个数据的子节点添加进队列，并出列添加至数组中
            while(size>0){
                //记录当前节点并出列
                TreeNode cur = q.poll();
                //把当前节点加入进数组中
                list.add(cur.val);
                //孩子节点非空的话，把孩子节点入列
                if(cur.left!=null){
                    q.add(cur.left);
                }
                if(cur.right!=null){
                    q.add(cur.right);
                }
                //每遍历一个数据size减一，当当前层size为0的时候代表着当前层遍历完了
                size--;
            }
            //这里新建一个数组的原因是，为了防止引用传递，拷贝一份list数组，添加进二维数组中
            result.add(new ArrayList<>(list));
        }
         //返回结果                                                                                                                     
        return result;
    }
}

class Solution {
    public List<List<Integer>> levelOrderBottom(TreeNode root) {
        List<List<Integer>> result =new ArrayList<>();
        if(root == null){
            return result; 
        }
        Queue<TreeNode> q = new LinkedList<>();
        q.add(root);
        //新建一个以链表存储数组，每个数组和数组直接以链表的形式链接
        LinkedList<ArrayList<Integer>> temp =new LinkedList<>();
        while(q.size()>0){
            int size = q.size();
            ArrayList<Integer> list = new ArrayList<>();
            while(size>0){
               TreeNode f = q.poll();
               list.add(f.val);
                if(f.left!=null){
                    q.add(f.left);
                }
                if(f.right!=null){
                    q.add(f.right); 
                }
                size--;
            }
            //把每一层的数组放在前面用 addFirst()函数
            temp.addFirst(new ArrayList<>(list));
            result=new ArrayList<>(temp);
        }
        return result;
    }
}