给定一个无序的数组,找出数组在排序之后,相邻元素之间最大的差值。
如果数组元素个数小于 2
,则返回 0
。
示例 1:
输入: [3,6,9,1]
输出: 3
解释: 排序后的数组是 [1,3,6,9], 其中相邻元素 (3,6) 和 (6,9) 之间都存在最大差值 3。
示例 2:
输入: [10]
输出: 0
解释: 数组元素个数小于 2,因此返回 0。
说明:
- 你可以假设数组中所有元素都是非负整数,且数值在 32 位有符号整数范围内。
- 请尝试在线性时间复杂度和空间复杂度的条件下解决此问题。
解答
首先基于排序,虽然能过,但显然不符合复杂度要求:
class Solution {
public:
int maximumGap(vector<int>& nums) {
if(nums.size() <= 1)
return 0;
sort(nums.begin(), nums.end());
int result = INT_MIN;
for(int i = 1; i < nums.size(); i++){
result = max(result, nums[i] - nums[i-1]);
}
return result;
}
};
实际上,说到线性复杂度,最容易想到的是桶排序,参考官方题解,有这样一个结论:排序后相邻数字的最大间隔不会小于(maxVal-minVal)/(n-1)
,(想象一下最极端的情况所有数字都相等),因此可以考虑划分若干个间隔为(maxVal-minVal)/(n-1)
,则排序后的最大间隔一定出现在相邻区间之间。实现时只需要记录每个区间内的最小值及最大值,然后相邻区间进行比较即可:
class Solution {
public:
int maximumGap(vector<int>& nums) {
int n = nums.size();
if(n <= 1)
return 0;
int minVal = *min_element(nums.begin(), nums.end());
int maxVal = *max_element(nums.begin(), nums.end());
// 最小值为1,避免除零
int bucketSize = max(1, (maxVal - minVal) / (n - 1));
int bucketCnt = (maxVal - minVal) / bucketSize + 1;
vector<pair<int, int>> buckets(bucketCnt, {-1, -1});
for(auto & num : nums){
int idx = (num - minVal) / bucketSize;
if(buckets[idx].first == -1){
buckets[idx].first = buckets[idx].second = num;
}
else{
buckets[idx].first = min(buckets[idx].first, num);
buckets[idx].second = max(buckets[idx].second, num);
}
}
int result = 0;
int prevIdx = -1;
for(int i = 0; i < bucketCnt; i++){
if(buckets[i].first == -1)
continue;
if(prevIdx != -1)
result = max(result, buckets[i].first - buckets[prevIdx].second);
prevIdx = i;
}
return result;
}
};