LeetCode 164. 最大间距

题目描述

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

如果数组元素个数小于 2，则返回 0。

示例 1:

输入: [3,6,9,1]
输出: 3
解释: 排序后的数组是 [1,3,6,9], 其中相邻元素 (3,6) 和 (6,9) 之间都存在最大差值 3。

示例 2:

输入: [10]
输出: 0
解释: 数组元素个数小于 2，因此返回 0。

说明:

• 你可以假设数组中所有元素都是非负整数，且数值在 32 位有符号整数范围内。
• 请尝试在线性时间复杂度和空间复杂度的条件下解决此问题。

问题分析

此题限制了O(n)的时间复杂度，所以我们用桶排序来做。设数组中有n个数，最大的数为maxn，最小的数为minn。那么每个桶的大小为(maxn - minn) / n + 1，桶的个数为(maxn - minn) / basket_size + 1。然后我们遍历数组nums，把每个数用式子(i - minn) / basket_size计算出应该放到哪个桶里，然后更新每个桶的最大值和最小值，然后将放的位置存入哈希set中，用来知道哪些桶不是空的。for循环结束后，数组basket_min就记录了每个桶的最小数，数组basket_max就记录了每个桶的最大数。由于桶的个数设定为(maxn - minn) / basket_size + 1，所以至少有一个空桶，如果某一个桶中有两个数，那就会有两个空桶了。由于每个桶的容量都一样，所以一个桶中的两个数的间距一定没有中间隔着空桶的两个数的间距大，因此下面我们利用哈希set找到非空桶，然后我们遍历非空桶，分别用每个非空桶的最小值与他的上一个非空桶的最大值计算间隔，遍历结束后所有间隔里取最大者就是数组nums的最大间隔。

代码实现

class Solution {
public:
    int maximumGap(vector<int>& nums) {
        if(nums.size() < 2)
            return 0;
        int n = nums.size();
        int minn = INT_MAX;
        int maxn = INT_MIN;
        for(int i = 0; i < n; i++){
            minn = min(minn, nums[i]);
            maxn = max(maxn, nums[i]);
        }
        int basket_size = (maxn - minn) / n + 1;
        int basket_num = (maxn - minn) / basket_size + 1;
        vector<int> basket_min(basket_num, INT_MAX);
        vector<int> basket_max(basket_num, INT_MIN);
        unordered_set<int> uset;
        for(int i : nums){
            int temp = (i - minn) / basket_size;
            basket_min[temp] = min(basket_min[temp], i);
            basket_max[temp] = max(basket_max[temp], i);
            uset.insert(temp);
        }
        int ans = 0;
        int pre = 0;
        for(int i = 1; i < basket_num; i++){
            if(!uset.count(i))
                continue;
            ans = max(ans, basket_min[i] - basket_max[pre]);
            pre = i;
        }
        return ans;
    }
};