代码随想录算法训练营第二十天|LeetCode530,501,236
530.二叉搜索树的最小绝对差
最简单的想法:通过中序遍历将二叉树转换成有序数组,然后遍历数组,获得最小值。
其实在二叉搜素树中序遍历的过程中,我们就可以直接计算了。需要用一个pre节点记录一下cur节点的前一个节点。这一点与上一题验证二叉搜索树的方式一样。
public class LeetCode530 {
TreeNode pre;
int min = Integer.MAX_VALUE;
public int getMinimumDifference(TreeNode root) {
inorder(root);
return min;
}
public void inorder(TreeNode root){
if (root == null){
return;
}
inorder(root.left);
if (pre!=null&&Math.abs(root.val-pre.val)<min){
min = Math.abs(root.val-pre.val);
}
pre = root;
inorder(root.right);
}
501二叉搜索树中的众数
考虑是二叉搜索树的情况,中序遍历是有序的
遍历有序数组的元素出现频率,从头遍历,那么一定是相邻两个元素作比较,然后就把出现频率最高的元素输出就可以了
在二叉搜索树上,可以考虑使用pre来记录前一个节点,与当前节点进行比较。
如果 频率count 等于 maxCount(最大频率),当然要把这个元素加入到结果集中,频率count 大于 maxCount的时候,不仅要更新maxCount,而且要清空结果
public class LeetCode501 {
TreeNode pre;
int count;
int max;
List<Integer> result;
public int[] findMode(TreeNode root) {
result = new ArrayList<>();
max = 0;
count = 0;
pre = null;
inorder(root);
return result.stream().mapToInt(Integer::intValue).toArray();
}
public void inorder(TreeNode node){
if (node == null){
return;
}
inorder(node.left);
int nodeValue = node.val;
//如果null为空,则当前判断的是第一个节点,初始化count为1
if (pre == null||nodeValue!= pre.val){
count = 1;
}else {
count++;
}
//更新结果以及maxCount
if(count>max){
result.clear();
result.add(node.val);
max = count;
} else if (count == max) {
result.add(node.val);
}
pre = node;
inorder(node.right);
}
}
236.二叉树的最近公共祖先
后序遍历(左右中)就是天然的回溯过程,可以根据左右子树的返回值,来处理中节点的逻辑。
首先最容易想到的一个情况:如果找到一个节点,发现左子树出现结点p,右子树出现节点q,或者 左子树出现结点q,右子树出现节点p,那么该节点就是节点p和q的最近公共祖先。
判断逻辑是 如果递归遍历遇到q,就将q返回,遇到p 就将p返回,那么如果 左右子树的返回值都不为空,说明此时的中节点,一定是q 和p 的最近祖先。(,题目强调:二叉树节点数值是不重复的,而且一定存在 q 和 p)
第二种情况:就是节点本身p(q),它拥有一个子孙节点q§,但实际上在实现第一种情况逻辑的时候,第二种情况也实现了。
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
if(root == null || root == p || root == q){
return root;
}
TreeNode left = lowestCommonAncestor(root.left,p,q);
TreeNode right = lowestCommonAncestor(root.right,p,q);
if (left!=null&&right!=null) return root;
if (left!=null&&right==null) {
return left;
}else if (left == null && right!=null){
return right;
}else {
return null;
}
}