leetcode - 2616. Minimize the Maximum Difference of Pairs

最新推荐文章于 2024-07-20 23:08:39 发布

KpLn_HJL

最新推荐文章于 2024-07-20 23:08:39 发布

阅读量366

点赞数

分类专栏： OJ题目记录文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/sinat_41679123/article/details/132180758

版权

OJ题目记录专栏收录该内容

515 篇文章 1 订阅

订阅专栏

Description

You are given a 0-indexed integer array nums and an integer p. Find p pairs of indices of nums such that the maximum difference amongst all the pairs is minimized. Also, ensure no index appears more than once amongst the p pairs.

Note that for a pair of elements at the index i and j, the difference of this pair is |nums[i] - nums[j]|, where |x| represents the absolute value of x.

Return the minimum maximum difference among all p pairs. We define the maximum of an empty set to be zero.

Example 1:

Input: nums = [10,1,2,7,1,3], p = 2
Output: 1
Explanation: The first pair is formed from the indices 1 and 4, and the second pair is formed from the indices 2 and 5. 
The maximum difference is max(|nums[1] - nums[4]|, |nums[2] - nums[5]|) = max(0, 1) = 1. Therefore, we return 1.

Example 2:

Input: nums = [4,2,1,2], p = 1
Output: 0
Explanation: Let the indices 1 and 3 form a pair. The difference of that pair is |2 - 2| = 0, which is the minimum we can attain.

Constraints:

1 <= nums.length <= 10^5
0 <= nums[i] <= 10^9
0 <= p <= (nums.length)/2

Solution

Think reversely, try to find a boundary, and see how many pairs we have that the distance is below or equals to boundary. If the number is larger than p, then we could shrink the boundary.

Use binary search, left = 0, right = max(nums) - min(nums).

Time complexity: $o(\log(\max(nums) - \min(nums) * n)$
Space complexity: $o (1)$

Code

class Solution:
    def minimizeMax(self, nums: List[int], p: int) -> int:
        nums.sort()
        left, right = 0, nums[-1] - nums[0]
        while left < right:
            mid = (left + right) >> 1
            # find how many pairs that are smaller than mid
            index = 0
            k = 0
            while index < len(nums) - 1:
                if abs(nums[index] - nums[index + 1]) <= mid:
                    k += 1
                    index += 2
                else:
                    index += 1
            if k >= p:
                right = mid
            else:
                left = mid + 1
        return (left + right) >> 1