题目描述(困难难度)

164. Maximum Gap - 图1

给一个乱序的数组,求出数组排序以后的相邻数字的差最大是多少。

解法一

先来个直接的,就按题目的意思,先排序,再搜索。

  1. public int maximumGap(int[] nums) {
  2. Arrays.sort(nums);
  3. int max = 0;
  4. for (int i = 0; i < nums.length - 1; i++) {
  5. if (nums[i + 1] - nums[i] > max) {
  6. max = nums[i + 1] - nums[i];
  7. }
  8. }
  9. return max;
  10. }

但是正常的排序算法时间复杂度会是 O(nlog(n)),题目要求我们在 O(n) 的复杂度下去求解。

自己想了各种思路,但想不出怎么降到 O(n),把 官方 给的题解分享一下吧。

我们一般排序算法多用快速排序,平均时间复杂度是 O(nlog(n)),其实还有一种排序算法,时间复杂度是 O(kn)k 是最大数字的位数,当 k 远小于 n 的时候,时间复杂度可以近似看成 O(n)。这种排序算法就是基数排序,下边讲一下具体思想。

比如这样一个数列排序: 342 58 576 356, 我们来看一下怎么排序:

  1. 不足的位数看做是 0
  2. 342 058 576 356
  3. 按照个位将数字依次放到不同的位置
  4. 0:
  5. 1:
  6. 2: 342
  7. 3:
  8. 4:
  9. 5:
  10. 6: 576, 356
  11. 7:
  12. 8: 058
  13. 9:
  14. 把上边的数字依次拿出来,组成新的序列 342 576 356 058,然后按十位继续放到不同的位置。
  15. 0:
  16. 1:
  17. 2:
  18. 3:
  19. 4: 342
  20. 5: 356 058
  21. 6:
  22. 7: 576
  23. 8:
  24. 9:
  25. 把上边的数字依次拿出来,组成新的序列 342 356 058 576,然后按百位继续装到不同的位置。
  26. 0: 058
  27. 1:
  28. 2:
  29. 3: 342 356
  30. 4:
  31. 5: 576
  32. 6:
  33. 7:
  34. 8:
  35. 9:
  36. 把数字依次拿出来,最终结果就是 58 342 356 576

为了代码更好理解, 我们可以直接用 10list 去存放每一组的数字,官方题解是直接用一维数组实现的。

对于取各个位的的数字,我们通过对数字除以 1, 10, 100… 然后再对 10 取余来实现。

  1. public int maximumGap(int[] nums) {
  2. if (nums.length <= 1) {
  3. return 0;
  4. }
  5. List<ArrayList<Integer>> lists = new ArrayList<>();
  6. for (int i = 0; i < 10; i++) {
  7. lists.add(new ArrayList<>());
  8. }
  9. int n = nums.length;
  10. int max = nums[0];
  11. //找出最大的数字
  12. for (int i = 1; i < n; i++) {
  13. if (max < nums[i]) {
  14. max = nums[i];
  15. }
  16. }
  17. int m = max;
  18. int exp = 1;
  19. //一位一位的进行
  20. while (max > 0) {
  21. //将之前的元素清空
  22. for (int i = 0; i < 10; i++) {
  23. lists.set(i, new ArrayList<>());
  24. }
  25. //将数字放入对应的位置
  26. for (int i = 0; i < n; i++) {
  27. lists.get(nums[i] / exp % 10).add(nums[i]);
  28. }
  29. //将数字依次拿出来
  30. int index = 0;
  31. for (int i = 0; i < 10; i++) {
  32. for (int j = 0; j < lists.get(i).size(); j++) {
  33. nums[index] = lists.get(i).get(j);
  34. index++;
  35. }
  36. }
  37. max /= 10;
  38. exp *= 10;
  39. }
  40. int maxGap = 0;
  41. for (int i = 0; i < nums.length - 1; i++) {
  42. if (nums[i + 1] - nums[i] > maxGap) {
  43. maxGap = nums[i + 1] - nums[i];
  44. }
  45. }
  46. return maxGap;
  47. }

解法二

上边的解法还不是真正的 O(n),下边继续介绍另一种解法,参考了 这里-time-and-space) ,评论区对自己帮助很多。

我们知道如果是有序数组的话,我们就可以通过计算两两数字之间差即可解决问题。

那么如果是更宏观上的有序呢?

  1. 我们把 0 3 4 6 23 28 29 33 38 依次装到三个箱子中
  2. 0 1 2 3
  3. ------- ------- ------- -------
  4. | 3 4 | | | | 29 | | 33 |
  5. | 6 | | | | 23 | | |
  6. | 0 | | | | 28 | | 38 |
  7. ------- ------- ------- -------
  8. 0 - 9 10 - 19 20 - 29 30 - 39
  9. 我们把每个箱子的最大值和最小值表示出来
  10. min max min max min max min max
  11. 0 6 - - 23 29 33 38

我们可以只计算相邻箱子 minmax 的差值来解决问题吗?空箱子直接跳过。

2 个箱子的 min 减第 0 个箱子的 max23 - 6 = 17

3 个箱子的 min 减第 2 个箱子的 max33 - 29 = 4

看起来没什么问题,但这样做一定需要一个前提,因为我们只计算了相邻箱子的差值,没有计算箱子内数字的情况,所以我们需要保证每个箱子里边的数字一定不会产生最大 gap

我们把箱子能放的的数字个数记为 interval,给定的数字中最小的是 min,最大的是 max。那么箱子划分的范围就是 min ~ (min + 1 * interval - 1)(min + 1 * interval) ~ (min + 2 * interval - 1)(min + 2 * interval) ~ (min + 3 * interval - 1)…,上边举的例子中, interval 我们其实取了 10

划定了箱子范围后,我们其实很容易把数字放到箱子中,通过 (nums[i] - min) / interval 即可得到当前数字应该放到的箱子编号。那么最主要的问题其实就是怎么去确定 interval

interval 过小的话,需要更多的箱子去存储,很费空间,此外箱子增多了,比较的次数也会增多,不划算。

interval 过大的话,箱子内部的数字可能产生题目要求的最大 gap,所以肯定不行。

所以我们要找到那个保证箱子内部的数字不会产生最大 gap,并且尽量大的 interval

继续看上边的例子,0 3 4 6 23 28 29 33 38,数组中的最小值 0 和最大值 38 ,并没有参与到 interval 的计算中,所以它俩可以不放到箱子中,还剩下 n - 2 个数字。

像上边的例子,如果我们保证至少有一个空箱子,那么我们就可以断言,箱子内部一定不会产生最大 gap

因为在我们的某次计算中,会跳过一个空箱子,那么得到的 gap 一定会大于 interval,而箱子中的数字最大的 gapinterval - 1

接下来的问题,怎么保证至少有一个空箱子呢?

鸽巢原理的变形,有 n - 2 个数字,如果箱子数多于 n - 2 ,那么一定会出现空箱子。总范围是 max - min,那么 interval = (max - min) / 箱子数,为了使得 interval 尽量大,箱子数取最小即可,也就是 n - 1

所以 interval = (max - min) / n - 1 。这里如果除不尽的话,我们 interval 可以向上取整。因为我们给定的数字都是整数,这里向上取整的话对于最大 gap 是没有影响的。比如原来范围是 [0,5.5),那么内部产生的最大 gap5 - 0 = 5。现在向上取整,范围变成[0,6),但是内部产生的最大 gap 依旧是 5 - 0 = 5

所有问题都解决了,可以安心写代码了。

  1. public int maximumGap(int[] nums) {
  2. if (nums.length <= 1) {
  3. return 0;
  4. }
  5. int n = nums.length;
  6. int min = nums[0];
  7. int max = nums[0];
  8. //找出最大值、最小值
  9. for (int i = 1; i < n; i++) {
  10. min = Math.min(nums[i], min);
  11. max = Math.max(nums[i], max);
  12. }
  13. if(max - min == 0) {
  14. return 0;
  15. }
  16. //算出每个箱子的范围
  17. int interval = (int) Math.ceil((double)(max - min) / (n - 1));
  18. //每个箱子里数字的最小值和最大值
  19. int[] bucketMin = new int[n - 1];
  20. int[] bucketMax = new int[n - 1];
  21. //最小值初始为 Integer.MAX_VALUE
  22. Arrays.fill(bucketMin, Integer.MAX_VALUE);
  23. //最小值初始化为 -1,因为题目告诉我们所有数字是非负数
  24. Arrays.fill(bucketMax, -1);
  25. //考虑每个数字
  26. for (int i = 0; i < nums.length; i++) {
  27. //当前数字所在箱子编号
  28. int index = (nums[i] - min) / interval;
  29. //最大数和最小数不需要考虑
  30. if(nums[i] == min || nums[i] == max) {
  31. continue;
  32. }
  33. //更新当前数字所在箱子的最小值和最大值
  34. bucketMin[index] = Math.min(nums[i], bucketMin[index]);
  35. bucketMax[index] = Math.max(nums[i], bucketMax[index]);
  36. }
  37. int maxGap = 0;
  38. //min 看做第 -1 个箱子的最大值
  39. int previousMax = min;
  40. //从第 0 个箱子开始计算
  41. for (int i = 0; i < n - 1; i++) {
  42. //最大值是 -1 说明箱子中没有数字,直接跳过
  43. if (bucketMax[i] == -1) {
  44. continue;
  45. }
  46. //当前箱子的最小值减去前一个箱子的最大值
  47. maxGap = Math.max(bucketMin[i] - previousMax, maxGap);
  48. previousMax = bucketMax[i];
  49. }
  50. //最大值可能处于边界,不在箱子中,需要单独考虑
  51. maxGap = Math.max(max - previousMax, maxGap);
  52. return maxGap;
  53. }

这道题主要是对排序算法的了解,第一次见到了基数排序的应用,解法二其实是桶排序的步骤。

windliang wechat

添加好友一起进步~

如果觉得有帮助的话,可以点击 这里 给一个 star 哦 ^^

如果想系统的学习数据结构和算法,强烈推荐一个我之前学过的课程,可以点击 这里 查看详情