LeetCode之Minimum Distance Between BST Nodes(Kotlin)

334 阅读1分钟

问题: Given a Binary Search Tree (BST) with the root node root, return the minimum difference between the values of any two different nodes in the tree.

Example :

Input: root = [4,2,6,1,3,null,null]
Output: 1
Explanation:
Note that root is a TreeNode object, not an array.

The given tree [4,2,6,1,3,null,null] is represented by the following diagram:

          4
        /   \
      2      6
     / \    
    1   3  

while the minimum difference in this tree is 1, it occurs between node 1 and node 2, also between node 3 and node 2.
Note:

The size of the BST will be between 2 and 100.
The BST is always valid, each node's value is an integer, and each node's value is different.

方法: 题目中的二叉搜索树节点间具有特殊的大小关系,即左节点<根节点<右节点,则通过中序遍历就是从小到大排序的节点,然后计算节点间的diff,取最小diff即为最终结果。

具体实现:

class MinimumDistanceBetweenBSTNodes {

    // Definition for a binary tree node.
    class TreeNode(var `val`: Int) {
        var left: TreeNode? = null
        var right: TreeNode? = null
    }

    private var pre = -100
    private var diff = 100

    fun minDiffInBST(root: TreeNode?): Int {
        if (root == null) {
            return Int.MIN_VALUE
        }
        minDiffInBST(root.left)
        diff = minOf(diff, root.`val` - pre)
        pre = root.`val`
        minDiffInBST(root.right)
        return diff
    }
}

fun main(args: Array<String>) {
    val tree4 = MinimumDistanceBetweenBSTNodes.TreeNode(4)
    val tree2 = MinimumDistanceBetweenBSTNodes.TreeNode(2)
    val tree6 = MinimumDistanceBetweenBSTNodes.TreeNode(6)
    val tree1 = MinimumDistanceBetweenBSTNodes.TreeNode(1)
    val tree3 = MinimumDistanceBetweenBSTNodes.TreeNode(3)
    tree4.left = tree2
    tree4.right = tree6
    tree2.left = tree1
    tree2.right = tree3
    val minimumDistanceBetweenBSTNodes = MinimumDistanceBetweenBSTNodes()
    println(minimumDistanceBetweenBSTNodes.minDiffInBST(tree4))
}

有问题随时沟通

具体代码实现可以参考Github