LeetCode 783. 二叉搜索树结点最小距离(Minimum Distance Between BST Nodes)

783. 二叉搜索树结点最小距离

LeetCode783. Minimum Distance Between BST Nodes

题目描述
给定一个二叉搜索树的根结点 root, 返回树中任意两节点的差的最小值。

示例:

输入: root = [4,2,6,1,3,null,null]
输出: 1
解释:
注意: root 是树结点对象 (TreeNode object),而不是数组。

给定的树 [4,2,6,1,3,null,null] 可表示为下图:

          4
        /   \
      2      6
     / \    
    1   3  

最小的差值是 1, 它是节点 1 和节点 2 的差值, 也是节点 3 和节点 2 的差值。

注意:

  • 二叉树的大小范围在 2 到 100。
  • 二叉树总是有效的,每个节点的值都是整数,且不重复。

Java 实现

import java.util.ArrayList;
import java.util.List;

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;

    TreeNode(int x) {
        val = x;
    }
}

public class Solution {
    // 中等效率
    private List<Integer> res = new ArrayList<>();

    public int minDiffInBST(TreeNode root) {
        if (root == null) {
            return 0;
        }
        handle(root);
        // Java 8 排序(由于题目是二叉排序树,中序遍历一棵二叉排序树的结点就可得到排好序的结点序列。)
        // res = res.stream().sorted(Integer::compareTo).collect(Collectors.toList());
        int min = Integer.MAX_VALUE;
        for (int i = 1; i < res.size(); i++) {
            min = Math.min(res.get(i) - res.get(i - 1), min);
        }
        return min;
    }

    // 二叉树中序遍历
    public void handle(TreeNode root) {
        if (root == null) {
            return;
        }
        handle(root.left);
        res.add(root.val);
        handle(root.right);
    }
}

参考资料

猜你喜欢

转载自www.cnblogs.com/hglibin/p/10817261.html