[桶排序]最大间隙问题-CSDN博客

2019独角兽企业重金招聘Python工程师标准>>>

这个问题很经典，解法很巧妙。如果是要求O(nlogn)的做法，那么只需要简单地排序一下就行了。但是这里要求O（n）,就需要另外的方法了。基于比较的排序方法，最好的也是O（nlogn），但是，还有一些不是基于比较的排序方法，例如桶排序。桶排序加速的思想为分治，将元素划分为若干个更小规模的桶，以后处理的时候，只需要处理同一个桶的元素。

说回这道题，这道题的思路也是用桶。假设有数组x[1...n]，其中最大的元素为max，最小元素为min，将左闭右开的实数区间[min，max)划分为n-1个等长的子区间（桶），每个子区间也是左闭右开的，我们用len来表示每个子区间的长度。除去max和min，剩下的n-2个数，每个数都属于其中一个桶。对于同一个桶的两个数，因为桶是左闭右开的，所以他们的距离肯定是小于len的。然后，关键的一点是，n-2个数放进n-1个桶，由抽屉原理可以知道，肯定有一个桶是空的，所以，距离最远的相邻的两个数，肯定是属于两个不同的桶。于是，我们可以把每个桶都扫描一次，相邻最远的两个数，必定其中一个是某个桶里的最大值，另一个是另一个桶里的最小值。

不过，事情还没完。刚才我们的做法是除去了max和min的，所以，最后我们还要考虑max和min对问题答案的影响，除了用上面的方法算出一个可能的答案外，我们还要考虑这两个数会不会是答案：“min 和第一个桶的最小值的距离”， “max 和最后一个桶的最大值的距离”。

The End

#include <vector>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cassert>
using namespace std;

template<class VType, class E>
void makeVector(VType & a, int len, E v) {
  a.resize(len);
  for (int i = 0; i < len; ++i) a[i] = v;
}


template<class Seq>
int findIndexOfMax(Seq a, int len) {
  int i;
  int index = 0;
  for (i = 0; i < len; ++i)
    if (a[i] > a[index]) index = i;
  return index;
}

template<class Seq>
int findIndexOfMin(Seq a, int len) {
  int i;
  int index = 0;
  for (i = 0; i < len; ++i)
    if (a[i] < a[index]) index = i;
  return index;
}


double maxGap(const vector<double> & a) {
  int n = a.size();
  int indexOfMax = findIndexOfMax(a, n);
  int indexOfMin = findIndexOfMin(a, n);

  double gap = (a[indexOfMax] - a[indexOfMin]) / (n - 1);
  assert(gap != 0);

  vector<double> bucketMax;
  vector<double> bucketMin;
  vector<int> bucketCount;

  makeVector(bucketMax, n - 1, -2000000000);
  makeVector(bucketMin, n - 1, 2000000000);
  makeVector(bucketCount, n - 1, 0);


  int i, j;
  for (i = 0; i < n; ++i) {
    if (i == indexOfMax || i == indexOfMin) continue;
    int bucketIndex = int(floor( (a[i] - a[indexOfMin]) / gap ));
    assert(bucketIndex >= 0 && bucketIndex < n - 1);
    bucketCount[bucketIndex] = bucketCount[bucketIndex] + 1;
    if (a[i] > bucketMax[bucketIndex]) bucketMax[bucketIndex] = a[i];
    if (a[i] < bucketMin[bucketIndex]) bucketMin[bucketIndex] = a[i];
  }


  double ans = -2000000000;
  i = 0;
  while (1) {
    while (i < n - 1 && bucketCount[i] == 0) ++i;  //从第一个桶开始往右边找第一个非空的桶
    j = i + 1;
    while (j < n - 1 && bucketCount[j] == 0) ++j;

    if (i < n - 1 && j < n - 1) {
      ans = max(ans, bucketMin[j] - bucketMax[i]);
      i = j;
    } else {
      break;
    }
  }

  i = 0;
  while (i < n - 1 && bucketCount[i] == 0) ++i;
  ans = max(ans, bucketMin[i] - a[indexOfMin]);

  i = n - 1 - 1;
  while (i >= 0 && bucketCount[i] == 0) --i;
  ans = max(ans, a[indexOfMax] - bucketMax[i]);

  return ans;
}


//=========================================
vector< double > x;

int main() {
  int N;
  scanf("%d", &N);
  x.resize(N);
  for (int i = 0; i < N; ++i) {
    scanf("%lf", &(x[i]));
  }

  printf("%lf\n", maxGap(x));

  return 0;
}

转载于:https://my.oschina.net/mustang/blog/58011