Binary Tree Level Order Traversal

28 篇文章 0 订阅

Given a binary tree, return the level order traversal of its nodes' values. (ie, from left to right, level by level).

For example:
Given binary tree {3,9,20,#,#,15,7},

    3
   / \
  9  20
    /  \
   15   7

return its level order traversal as:

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

confused what "{1,#,2,3}" means?

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public ArrayList<ArrayList<Integer>> levelOrder(TreeNode root) {
        // Start typing your Java solution below
        // DO NOT write main() function
		ArrayList<Integer> inner_list = new ArrayList<Integer>();
		ArrayList<ArrayList<Integer>> outer_list = new ArrayList<ArrayList<Integer>>();
		
		if(root == null){
			return outer_list;
		}
		
		Queue<TreeNode> current = new LinkedList<TreeNode>();
		Queue<TreeNode> next = new LinkedList<TreeNode>();
		TreeNode temp;
		current.offer(root);
		while(!current.isEmpty()){
			temp = current.poll();
			inner_list.add(temp.val);
			if(temp.left!=null){
				next.offer(temp.left);
			}
			if(temp.right!=null){
				next.offer(temp.right);
			}
			if(current.isEmpty()){
				outer_list.add(inner_list);
				inner_list = new ArrayList<Integer>();
				// wrong using inner_list.clear()
				// clear() will clear the content in heap
				// elements in outer_list will also get cleaned up
				Queue<TreeNode> temp_queue = current;
				current = next;
				next = temp_queue;
			}
		}
		return outer_list;
    }
}

Second method:

using two indexes to track within one queue

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    // use two index to track positions within one queue
	// "first" is used to measure if there is any elements left in current level
	// "second" is used to measure how many elements in current level
	public ArrayList<ArrayList<Integer>> levelOrder(TreeNode root) {
		ArrayList<Integer> inner_list = new ArrayList<Integer>();
		ArrayList<ArrayList<Integer>> outer_list = new ArrayList<ArrayList<Integer>>();
		
		if(root == null){
			return outer_list;
		}
		
		Queue<TreeNode> queue = new LinkedList<TreeNode>();
		TreeNode temp;
		queue.offer(root);
		int first = 1;
		int second = 0;
		while(!queue.isEmpty()){
			temp = queue.poll();
			first--;
			inner_list.add(temp.val);
			if(temp.left!=null){
				queue.offer(temp.left);
				second++;
			}
			if(temp.right!=null){
				queue.offer(temp.right);
				second++;
			}
			if(first==0){
				first=second;
				second=0;
				outer_list.add(inner_list);
				inner_list = new ArrayList<Integer>();
			}
		}
		return outer_list;
	}
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值