LeetCode-day26-2740. 找出分区值

题目描述

给你一个 正 整数数组 nums 。

将 nums 分成两个数组：nums1 和 nums2 ，并满足下述条件：

• 数组 nums 中的每个元素都属于数组 nums1 或数组 nums2 。
• 两个数组都 非空 。
• 分区值 最小 。

分区值的计算方法是 |max(nums1) - min(nums2)| 。

其中，max(nums1) 表示数组 nums1 中的最大元素，min(nums2) 表示数组 nums2 中的最小元素。

返回表示分区值的整数。

示例

示例1：

输入：nums = [1,3,2,4]
输出：1
解释：可以将数组 nums 分成 nums1 = [1,2] 和 nums2 = [3,4] 。
-数组 nums1 的最大值等于 2 。
-数组 nums2 的最小值等于 3 。
分区值等于 |2 - 3| = 1 。
可以证明 1 是所有分区方案的最小值。

示例2：

输入：nums = [100,1,10]
输出：9
解释：可以将数组 nums 分成 nums1 = [10] 和 nums2 = [100,1] 。
-数组 nums1 的最大值等于 10 。
-数组 nums2 的最小值等于 1 。
分区值等于 |10 - 1| = 9 。
可以证明 9 是所有分区方案的最小值。

思路

贪心策略，步骤如下：

1. 把 nums 从小到大排序。
2. 排序后，最小值必然对应两个相邻元素，设其为 nums[i−1] 和 nums[i]。
3. 把 ≤nums[i−1] 的数分到第一个数组中，把 ≥nums[i] 的数分到第二个数组中。

代码

class Solution:
    def findValueOfPartition(self, nums: List[int]) -> int:
        nums.sort()
        n = len(nums)
        ans = inf
        for i in range(1, n):
            ans = min(ans,nums[i] - nums[i-1])
        return ans