501. Find Mode in Binary Search Tree
總是先算
```java
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
maxCnt
currCnt
prev
result
if (currCnt > maxCnt) {
result.clear();
result.add(node.val);
} else if (currCnt == maxCnt) {
result.add(node.val);
}
1 2 1 2 3 1 2 3
1 1 2 2 2 3 3 3
*/
class Solution {
int maxCnt = 0; // 2
int currCnt = 1; // 2
Integer prev = null; //2
// result [2]
public int[] findMode(TreeNode root) {
List<Integer> result = new ArrayList<>();
dfs(root, result);
return result.stream().mapToInt(i -> i).toArray();
}
private void dfs(TreeNode node, List<Integer> result) {
if (node == null) {
return;
}
dfs(node.left, result);
if (prev != null) {
if (prev != node.val) {
currCnt = 1;
} else {
currCnt++;
}
}
if (currCnt > maxCnt) {
maxCnt = currCnt;
result.clear();
result.add(node.val);
} else if (currCnt == maxCnt) {
result.add(node.val);
}
prev = node.val;
dfs(node.right, result);
}
}
```後算
Last updated
Was this helpful?