题目
链接:https://leetcode-cn.com/problems/maximum-gap
给定一个无序的数组,找出数组在排序之后,相邻元素之间最大的差值。
如果数组元素个数小于 2,则返回 0。
示例 1:
输入: [3,6,9,1]
输出: 3
解释: 排序后的数组是 [1,3,6,9], 其中相邻元素 (3,6) 和 (6,9) 之间都存在最大差值 3。
示例 2:
输入: [10]
输出: 0
解释: 数组元素个数小于 2,因此返回 0。
说明:
你可以假设数组中所有元素都是非负整数,且数值在 32 位有符号整数范围内。
请尝试在线性时间复杂度和空间复杂度的条件下解决此问题。
桶排序
- 确定桶的长度
由反证法可以证明相邻元素的间距不大于
c
e
i
l
(
(
m
a
x
V
a
l
−
m
i
n
V
a
l
)
/
(
n
−
1
)
)
ceil((maxVal - minVal) / (n-1))
ceil((maxVal−minVal)/(n−1))所以可以将整个区间minVal ~ maxVal
划分为若干个大小为
d
=
f
l
o
o
r
(
(
m
a
x
V
a
l
−
m
i
n
V
a
l
)
/
(
n
−
1
)
)
d = floor((maxVal - minVal) / (n-1))
d=floor((maxVal−minVal)/(n−1))的区间,这样相邻的元素一定不会落在同一个区间(桶)中,所以桶的数量为
(
m
a
x
V
a
l
−
m
i
n
V
a
l
)
/
d
+
1
(maxVal - minVal)/d + 1
(maxVal−minVal)/d+1。找到每个元素所在的桶后只需要维护每个桶中元素的最大值与最小值,然后从前到后不断比较相邻的桶,用当前桶的最小值减去前一个桶的最大值来计算间距。
class Solution {
public int maximumGap(int[] nums) {
int n = nums.length;
if (n < 2) return 0;
int maxVal = Arrays.stream(nums).max().getAsInt();
int minVal = Arrays.stream(nums).min().getAsInt();
// 相邻数字的最大间距不会小于ceil((max-min)/(n-1)), d最小值为1
int d = Math.max(1, (maxVal - minVal) / (n - 1));
int bucketSize = (maxVal - minVal) / d + 1; // 每个桶的大小
int[][] bucket = new int[bucketSize][2];
for (int i = 0; i < bucketSize; i++) {
Arrays.fill(bucket[i], -1);
}
for (int i = 0; i < n; i++) {
int idx = (nums[i] - minVal) / d;
if (bucket[idx][0] == -1) {
bucket[idx][0] = bucket[idx][1] = nums[i];
} else {
bucket[idx][0] = Math.min(bucket[idx][0], nums[i]);
bucket[idx][1] = Math.max(bucket[idx][1], nums[i]);
}
}
int ret = 0;
int prev = 0;
for (int i = 1; i < bucketSize; i++) {
if (bucket[i][0] == -1) {
continue;
}
ret = Math.max(ret, bucket[i][0] - bucket[prev][1]);
prev = i;
}
return ret;
}
}