leetcode102:二叉树的层序遍历

给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。

在这里插入图片描述

思路:可以利用队列先进先出的特性,将将二叉树中的节点按层次放进队列中,然后再依次出队列。
具体代码如下:

/**
 * 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) {
        List<List<Integer>> ret = new ArrayList<>();
        if(root==null) {
            return ret;
        }
        Queue<TreeNode> q = new LinkedList<>();
        q.offer(root);         //先将根节点放入队列中
        while(!q.isEmpty()) {  //然后开始循环遍历
            int size = q.size(); //size为当前层的元素个数
            List<Integer> level = new ArrayList<>(size);//用level来保存一层元素
            //一次性遍历一层的元素
            for(int i=0;i<size;i++) {
                TreeNode cur = q.poll();  //将当前层节点依次出队列
                level.add(cur.val);     //将当前层节点值放进level中
                //如果节点有左孩子,就将左孩子入队列
                if(cur.left != null) {
                    q.offer(cur.left);
                }
                //如果节点有右孩子,就将右孩子入队列
                if(cur.right != null) {
                    q.offer(cur.right);
                }
            }
            ret.add(level);   //每遍历完一层就将当前层结果放入ret中
        }
        return ret;
    }
}
发布了27 篇原创文章 · 获赞 57 · 访问量 938

猜你喜欢

转载自blog.csdn.net/A_K_L/article/details/104714757