二叉树的层次遍历

题目1

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

例如:
给定二叉树: [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回其层次遍历结果:

[
  [3],
  [9,20],
  [15,7]
]

解题思路:使用广度优先遍历,子循环使用层次节点数递减
代码如下:

public List<List<Integer>> levelOrder(TreeNode root) {
        //广度优先遍历,加入所属层级
		List<List<Integer>> res = new ArrayList<List<Integer>>();
		if(root == null) {
			return res;
		}
		LinkedList<TreeNode> queue = new LinkedList<TreeNode>();
		queue.offer(root);
		while(!queue.isEmpty()) {
			//每次遍历一层节点
			List<Integer> list = new ArrayList<Integer>();
			int count = queue.size();
			while(count-- > 0) {
				TreeNode cNode = queue.poll();
				list.add(cNode.val);
				if(cNode.left != null) {
					queue.offer(cNode.left);
				}
				if(cNode.right != null) {
					queue.offer(cNode.right);
				}
			}
			res.add(list);
		}
		return res;
    }

题目2:

给定一个二叉树,返回其节点值自底向上的层次遍历。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)

例如:
给定二叉树 [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回其自底向上的层次遍历为:

[
  [15,7],
  [9,20],
  [3]
]

代码实现

public List<List<Integer>> levelOrderBottom(TreeNode root) {
		List<List<Integer>> list = new ArrayList<List<Integer>>();
		if (root==null) {
			return list;
		}
		TreeNode node = root;
		Queue<TreeNode> queue = new LinkedList<TreeNode>();
		queue.offer(node);
		int levelCount=1;
		while(!queue.isEmpty()) {
			List<Integer> level = new LinkedList<Integer>();
			levelCount = queue.size();
			while(levelCount -->0) {
				TreeNode tmp = queue.poll();
				level.add(tmp.val);
				if (tmp.left!= null) {
					queue.offer(tmp.left);
				}
				if(tmp.right!=null) {
					queue.offer(tmp.right);
				}
			}
			list.add(level);
		}
		Collections.reverse(list);
		return list;
    }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值