LeetCode 164. 最大间距

LeetCode 

164. 最大间距

因为追求线性复杂度，所以不可能用经典排序的算法，因此需要更巧妙地利用已知条件。 类似与抽屉原理。我们用O(n)预处理出数组的最大值maxn和最小值minn，那么最大的间距肯定在maxn-minn之间，有因为有n个数，所以每个数之间的平均距离为size=(maxn-minn)/n+1, 这是如果数组有序且元素都是等间距的时候的最大间距。因此我们一共要把数组按每个元素与最小值minn的间距大小分为(maxn-minn)/size+1个小组。每个组中保存组里面的最大值和最小值。那么对最后分到数的分组来说，整体的最大间距就是每个有效的分组的最小值减去前一个分组的最大值。

class Solution {
public:
    int maximumGap(vector<int>& nums) {
        int n=nums.size();
        if(n<2)return 0;
        int ans=0;
        int maxn=0;
        for(int i=0;i<n;i++)maxn=max(maxn,nums[i]);
        int minn=maxn;
        for(int i=0;i<n;i++)minn=min(minn,nums[i]);
        int sz=(maxn-minn)/n+1;
        int bucket_n=(maxn-minn)/sz+1;
        vector<int>bmin(bucket_n,maxn+1),bmax(bucket_n,minn-1);
        for(int i=0;i<n;i++)
        {
            int idx=(nums[i]-minn)/sz;
            bmin[idx]=min(bmin[idx],nums[i]);
            bmax[idx]=max(bmax[idx],nums[i]);
        }
        for(int i=0;i<bucket_n;i++)
        {
            while(bmax[i]==minn+1)i++;
            int j=i+1;
            if(j>=bucket_n)break;
            while(bmin[j]==maxn+1)j++;
            ans=max(ans,bmin[j]-bmax[i]);
            i=j-1;
        }
        return ans;
    }
};