164. 最大间距
给定一个无序的数组,找出数组在排序之后,相邻元素之间最大的差值。
如果数组元素个数小于 2,则返回 0。
示例 1:
输入: [3,6,9,1] 输出: 3 解释: 排序后的数组是 [1,3,6,9], 其中相邻元素 (3,6) 和 (6,9) 之间都存在最大差值 3。
示例 2:
输入: [10] 输出: 0 解释: 数组元素个数小于 2,因此返回 0。
说明:
- 你可以假设数组中所有元素都是非负整数,且数值在 32 位有符号整数范围内。
- 请尝试在线性时间复杂度和空间复杂度的条件下解决此问题。
思路:
如果题目没有时空限制,直接排序即可。
但是题目要求使用线性时间和空间,所以不能排序。
考虑使用桶的思想,将每个数划分到对应的桶中,使得最大值只能是两个不同桶种的数相减。
问题是桶的大小如何设置?
- 找到最大值
max
和最小值min
- 假如排好序后,两个相邻的数之间会有一个差值,个数有
n - 1
个,等价于a1 - a0 + a2 - a1 + ... = max - min
- 如果每个桶能存
x
个数,则桶内最大值减最小值最大为x - 1
,若最终结果不为桶内两个元素相减,应满足(n - 1) * (x - 1) < max - min
,解得x <= (max - min - 1) / (n - 1) + 1
- 确定桶的大小为x,则桶的个数为
(max - min + 1 + x - 1) / x
(这里有一个下取整转上取整的操作)
之后就能遍历数组,确定每一个数映射到哪一个桶,然后更新对应桶的最大值和最小值。
最终结果为不为空的相邻桶(因为有的桶可能没有数,需跳过)的大桶的最小值减去小桶的最大值。
class Solution {
public int maximumGap(int[] nums) {
if (nums.length < 2)
return 0;
int max = -1, min = Integer.MAX_VALUE;
for (int x : nums) {
max = Math.max(max, x);
min = Math.min(min, x);
}
int n = nums.length;
int x = (max - min - 1) / (n - 1) + 1;
int len = (max - min + 1 + x - 1) / x;
int[][] a = new int[len][2];
for (int i = 0; i < len; i++) {
a[i][0] = Integer.MAX_VALUE;
a[i][1] = -1;
}
for (int v : nums) {
int idx = (v - min) / x;
a[idx][0] = Math.min(a[idx][0], v);
a[idx][1] = Math.max(a[idx][1], v);
}
max = -1;
for (int i = 0; i < len; i++) {
int j = i + 1;
while (j < len && a[j][1] == -1)
j++;
if (j == len) break;
max = Math.max(max, a[j][0] - a[i][1]);
i = j - 1;
}
return max == -1 ? 0 : max;
}
}
类似的问题还有:
存在重复元素III
5969. 摧毁小行星