leetcode “515. find largest value in each tree row”

LeetCode link

first thought

  • DFS

DFS solution

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
public class Solution {
public List<Integer> largestValues(TreeNode root) {
List<Integer> result = new ArrayList<>();
this.largestValues(root, 0, result);
return result;
}
private void largestValues(TreeNode node, int depth, List<Integer> result) {
if (node == null) {
return;
}
if (result.size() <= depth) {
result.add(node.val);
}
else if (result.get(depth) < node.val) {
result.set(depth, node.val);
}
this.largestValues(node.left, depth + 1, result);
this.largestValues(node.right, depth + 1, result);
}
}

BFS solution

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
class Solution {
public List<Integer> largestValues(TreeNode root) {
List<Integer> result = new ArrayList<>();
if (root == null) {
return result;
}
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
while (!queue.isEmpty()) {
int max = Integer.MIN_VALUE;
Queue<TreeNode> queue2 = new LinkedList<>();
while (!queue.isEmpty()) {
TreeNode node = queue.poll();
max = Math.max(max, node.val);
if (node.left != null) {
queue2.add(node.left);
}
if (node.right != null) {
queue2.add(node.right);
}
}
result.add(max);
queue = queue2;
}
return result;
}
}