欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

(Java)leetcode-530 Minimum Absolute Difference in BST (二叉搜索树的最小绝对差)

程序员文章站 2022-03-07 18:21:13
...

题目描述

给你一棵所有节点为非负值的二叉搜索树,请你计算树中任意两节点的差的绝对值的最小值。
(Java)leetcode-530 Minimum Absolute Difference in BST (二叉搜索树的最小绝对差)

思路1

先中序遍历将结果存为list,再遍历一遍,只在相邻的两个值之间求差值(二叉搜索树的性质,决定了相邻的值差值最小),并更新差值的最小值。

代码

class Solution {
	public int min = Integer.MAX_VALUE;
	public List<Integer> list = new ArrayList<>();
    public int getMinimumDifference(TreeNode root) {
    	travel(root);
    	return findMin();
    }

    public void travel(TreeNode root) {
    	if(root == null) return;    
        travel(root.left);
        list.add(root.val);
        travel(root.right);
    }

    public int findMin() {
    	for (int i = 1; i < list.size(); i++) {
    		min = Math.min(list.get(i) - list.get(i-1), min);
    	}
    	return min;
    }
}

(Java)leetcode-530 Minimum Absolute Difference in BST (二叉搜索树的最小绝对差)
此做法效率不高,做了两次遍历,且空间复杂度为O(n)。

改进

不必记录到数组里,因为每次只需要比较相邻的值,那么维持一个索引,始终指向(中序遍历的)前一个值,每次比较当前值和索引值并进行更新即可。 时间复杂度 O(N), 空间复杂度 O(1)。

代码

public class Solution {
    int min = Integer.MAX_VALUE;
    Integer prev = null;
    
    public int getMinimumDifference(TreeNode root) {
        if (root == null) return min;
        
        getMinimumDifference(root.left);
        
        if (prev != null) {
            min = Math.min(min, root.val - prev);
        }
        // 更新索引
        prev = root.val;
        
        getMinimumDifference(root.right);
        
        return min;
    }
    
}

(Java)leetcode-530 Minimum Absolute Difference in BST (二叉搜索树的最小绝对差)