难度:easy
一道很考细节的简单题
/**
* 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;
* }
* }
*/
class Solution {
public int[] findMode(TreeNode root) {
Stack<TreeNode> stack = new Stack<>();
List<Integer> list = new ArrayList<>();
TreeNode pre = null;
TreeNode cur = root;
int count = 0;
int maxCount = 0;
while (cur != null || !stack.isEmpty()) {
if (cur != null) {
stack.push(cur);
cur = cur.left;
} else {
cur = stack.pop();
if (pre == null || pre.val != cur.val) {
count = 1;
} else if (pre.val == cur.val){
count++;
}
if (count > maxCount) {
maxCount = count;
// count = 0;
list.clear();
list.add(cur.val);
} else if (count == maxCount) {
list.add(cur.val);
}
pre = cur;
cur = cur.right;
}
}
int[] ans = new int[list.size()];
for (int i = 0; i < list.size(); i++) {
ans[i] = list.get(i);
}
return ans;
}
}