0
点赞
收藏
分享

微信扫一扫

二叉树 | 层序遍历

杰森wang 2022-04-30 阅读 34
算法

二叉树的层序遍历

题目

给你二叉树的根节点root,返回二叉树的层序遍历序列。

思路:迭代

  • 数据结构:队列
  • 从根节点开始,先将跟根结点入队
  • 依次弹出队列中的结点,将当前结点的左孩子和右孩子分别入队
  • 直到遍历到最后一个节点

代码

    public static List<List<Integer>> levelOrder(TreeNode root) {//层序遍历
    	List<List<Integer>> res = new LinkedList<List<Integer>>();
    	if(root == null) return res;//根节点不为空
    	
    	Queue<TreeNode> que = new LinkedBlockingQueue<TreeNode>();//当前层的队列
    	que.add(root);//队列初始化
    	
    	while(!que.isEmpty()) {//当队列不为空,即上一层有结点
    		int last_size = que.size();//上一层的个数-que的长度会随着poll()变化,所以要记录
        	List<Integer> ceng = new LinkedList<Integer>();//标记上一层的结点
        	
    		for(int i = 0; i < last_size; i++) {//对上一层的每一个节点
    	    	TreeNode cur = new TreeNode();//当前操作的节点
    	    	cur = que.poll();//上一层的节点还在队列中
    	    	ceng.add(cur.val);//加入当前层
    	    	
    			if(cur.left != null) {//上一层节点的左结点入队
    				que.add(cur.left);
    			}
    			if(cur.right != null) {//上一层的右结点入队
    				que.add(cur.right);
    			}
    		}//for
    		
    		res.add(ceng);//把上一层加入结果list
    	}//while
    	return res;
    }//levelOrder
举报

相关推荐

0 条评论