Leetcode107. 二叉树的层次遍历 II
题目:
给定一个二叉树,返回其节点值自底向上的层次遍历。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)
例如:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其自底向上的层次遍历为:
[
[15,7],
[9,20],
[3]
]
题解:
scala代码如下:
迭代:
def levelOrderBottom(root: TreeNode): util.ArrayList[util.ArrayList[Int]] = {
val result: util.ArrayList[util.ArrayList[Int]] = new util.ArrayList[util.ArrayList[Int]]()
var flag = true
if (root == null) {
flag = false
}
if (flag) {
val list = new util.LinkedList[TreeNode]()
list.add(root)
while (!list.isEmpty) {
val temp = new util.ArrayList[Int]()
for (i <- 0 until list.size()) {
val node = list.poll()
temp.add(node.value)
if (node.left != null) list.add(node.left)
if (node.right != null) list.add(node.right)
}
println(temp.toArray.toBuffer)
result.add(temp)
}
Collections.reverse(result)
}
result
}
递归:
/**
* 递归
*
* @param root
* @return
*/
def levelOrderBottom2(root: TreeNode): util.ArrayList[util.ArrayList[Int]] = {
val result = new util.ArrayList[util.ArrayList[Int]]()
helper(root, 0, result)
Collections.reverse(result)
result
}
def helper(root: TreeNode, depth: Int, result: util.ArrayList[util.ArrayList[Int]]): Unit = {
var flag = true
if (root == null) {
flag = false
}
if (flag) {
if (depth + 1 > result.size()) {
result.add(new util.ArrayList[Int]())
}
result.get(depth).add(root.value)
if (root.left != null) helper(root.left, depth + 1, result)
if (root.right != null) helper(root.right, depth + 1, result)
}
}
java代码:
/**
*
* @param root
* @return
*/
public static List<List<Integer>> levelOrderBottom(TreeNode root) {
if (root == null) return null;
List<List<Integer>> res = new ArrayList<>();
Queue<TreeNode> q = new LinkedList<>();
q.add(root);
while (!q.isEmpty()) {
List<Integer> list = new ArrayList<>();
int len = q.size();
for (int i = 0; i < len; i++) {
TreeNode node = q.poll();
list.add(node.value);
if (node.left != null) q.add(node.left);
if (node.right != null) q.add(node.right);
}
System.out.println("----" + list.size());
res.add(list);
}
Collections.reverse(res);
return res;
}