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 位有符号整数范围内。
- 请尝试在线性时间复杂度和空间复杂度的条件下解决此问题。
题解
桶
我们可以利用桶将这些数字放进桶中。我们先求数组中的最大值和最小值,如果最大值和最小值相差小于2,则直接返回这个差值即可。其他条件下,这个数属于哪个桶,我们就用数-最小值,然后除以桶的大小。桶的大小是最大值-最小值,然后除以数的数量。显然,最终答案就出于,相邻桶中,最小值-最大值。详细过程见代码
代码
int maximumGap(vector<int>& nums) {
if(nums.empty()) return 0;
int maxNum=0,minNum=INT_MAX,n=nums.size();
if(n == 1) return 0;
else if(n == 2) return fabs(nums[0]-nums[1]);
for(int i=0; i<n; i++){ //求数组最大值和最小值
maxNum = max(maxNum,nums[i]);
minNum = min(minNum,nums[i]);
}
if(maxNum - minNum <= 1) return maxNum-minNum;
int size = (maxNum-minNum)/n; //桶的个数
if(size<1) size=1;
int num = (maxNum-minNum)/size + 1;
vector<int> maxPail(num,0);
vector<int> minPail(num,INT_MAX);
vector<bool> use(num,false);
for(int i=0; i<n; i++){
int p = (nums[i] - minNum)/size; //所属桶的下标
maxPail[p] = max(maxPail[p],nums[i]);
minPail[p] = min(minPail[p],nums[i]);
use[p] = true; //表明这个桶有都数
}
int ans=0;
int last=0;
for(int i=1; i<num; i++){
if(!use[i]) continue;
ans = max(ans,minPail[i]-maxPail[last]);
last = i;
}
return ans;
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-gap
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。