題目地址:
https://www.lintcode.com/problem/find-largest-value-in-each-tree-row/description
給定一棵二叉樹,返回其每層最大值。
法1:BFS。分層遍歷,記錄最大值加入最終結果即可。代碼如下:
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;
public class Solution {
/**
* @param root: a root of integer
* @return: return a list of integer
*/
public List<Integer> largestValues(TreeNode root) {
// write your code here
List<Integer> res = new ArrayList<>();
if (root == null) {
return res;
}
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
while (!queue.isEmpty()) {
int size = queue.size();
int max = Integer.MIN_VALUE;
for (int i = 0; i < size; i++) {
TreeNode x = queue.poll();
max = Math.max(max, x.val);
if (x.left != null) {
queue.offer(x.left);
}
if (x.right != null) {
queue.offer(x.right);
}
}
res.add(max);
}
return res;
}
}
class TreeNode {
int val;
TreeNode left, right;
TreeNode(int x) {
val = x;
}
}
時空複雜度。
法2:DFS。對整個樹做DFS,同時將深度作爲參數傳遞下去。如果當前深度是第一次達到,則將當前節點值直接加入res,否則比較之前遍歷時得到的最大值和當前節點值哪個大,再set回res即可。代碼如下:
import java.util.ArrayList;
import java.util.List;
public class Solution {
/**
* @param root: a root of integer
* @return: return a list of integer
*/
public List<Integer> largestValues(TreeNode root) {
// write your code here
List<Integer> res = new ArrayList<>();
dfs(root, 0, res);
return res;
}
private void dfs(TreeNode root, int depth, List<Integer> res) {
if (root == null) {
return;
}
// 如果depth是第一次達到,則將當前值加入res;
// 否則更新當前深度的最大值
if (depth == res.size()) {
res.add(root.val);
} else {
res.set(depth, Math.max(res.get(depth), root.val));
}
dfs(root.left, depth + 1, res);
dfs(root.right, depth + 1, res);
}
}
時間複雜度,空間。