leetcode164. 最大间距

给定一个无序的数组,找出数组在排序之后,相邻元素之间最大的差值。

如果数组元素个数小于 2,则返回 0。
示例 1:
输入: [3,6,9,1]
输出: 3
解释: 排序后的数组是 [1,3,6,9], 其中相邻元素 (3,6) 和 (6,9) 之间都存在最大差值 3。
示例 2:
输入: [10]
输出: 0
解释: 数组元素个数小于 2,因此返回 0。
说明:
你可以假设数组中所有元素都是非负整数,且数值在 32 位有符号整数范围内。
请尝试在线性时间复杂度和空间复杂度的条件下解决此问题。

要用线性的时间复杂度和空间复杂度,所以想到了桶排序。桶排序根据元素区间将元素放进一个个桶里面分别排序,主要适用于小范围整数数据,且数据独立均匀分布。步骤:1.设置固定数量的桶;2.将数据放入桶中;3.对每个桶排序;4.合并得到最终排序结果。

class Solution:
    def maximumGap(self, nums: List[int]) -> int:
        if len(nums) < 2:
            return 0
        min_num, max_num, len_num = min(nums), max(nums), len(nums)
        cap = (max_num-min_num)//len_num + 1  # 每个桶容量
        bucket_n = (max_num-min_num)//cap + 1  # 桶个数
        if bucket_n == 1:
            return max_num-min_num
        res = [[max_num+1, min_num-1] for _ in range(bucket_n)]  # 每个桶的最小最大值
        for num in nums:  # 记录每个桶的最大最小值
            loc = (num-min_num) // cap
            res[loc][0] = min(num, res[loc][0])
            res[loc][1] = max(num, res[loc][1])
        ret, before_right = 0, max_num+1
        for one in res:  # 最大间距一定出现在两桶之间
            if one[0] == max_num+1:
                continue
            if before_right != max_num+1:
                ret = max(ret, one[0]-before_right)
            before_right = one[1]
        return ret
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章