Leetcode刷题笔记 164. 最大间距

时间：2020年11月26日
知识点：桶排序、基数排序
题目链接：https://leetcode-cn.com/problems/reorganize-string/

题目
给定一个无序的数组，找出数组在排序之后，相邻元素之间最大的差值。

如果数组元素个数小于 2，则返回 0。

示例 1:
输入: [3,6,9,1]
输出: 3
解释: 排序后的数组是 [1,3,6,9], 其中相邻元素 (3,6) 和 (6,9) 之间都存在最大差值 3。

示例 2:
输入: [10]
输出: 0
解释: 数组元素个数小于 2，因此返回 0。

说明:

• 你可以假设数组中所有元素都是非负整数，且数值在 32 位有符号整数范围内。
• 请尝试在线性时间复杂度和空间复杂度的条件下解决此问题。

解法：

1. 这道题困难的是 在线性时间复杂度和空间复杂度内解决此问题
2. 用桶排序 降低复杂度
3. 这里如何设置桶的宽度呢 可以通过反证法 证明 相邻数字的最大间距不会小于 (max-min)/(n-1)
4. 这里我们向下取整 就可以得到初步桶宽 从而得到要几个桶
5. 遍历数据 找到对应的桶 更新桶的宽度
6. 最后比较各个桶间的距离即可
7. 后一种方法使用了 基数排序

代码

#include <stdio.h>
#include <iostream>
#include <vector>
using namespace std;
class Solution {
public:
    int maximumGap(vector<int>& nums) {
        int n = nums.size();
        if(n < 2) return 0;
        int maxN = *max_element(nums.begin(), nums.end());
        int minN = *min_element(nums.begin(), nums.end());
        int d = max(1,(maxN-minN)/(n-1));
        int bucketsize = (maxN-minN)/d + 1;
        
        vector<pair<int,int>> bucket(bucketsize,{-1,-1});
        for(int i = 0;i < n;i++){
            int idx = (nums[i]-minN)/d;
            if(bucket[idx].first == -1)
                bucket[idx].first = bucket[idx].second = nums[i];
            else{
                bucket[idx].first = min(nums[i],bucket[idx].first);
                bucket[idx].second = max(nums[i],bucket[idx].second);
            }
        }
        
        int pre = -1;
        int ans  = 0;
        for(int i = 0;i < bucketsize;i++){
            if(bucket[i].first==-1)
                continue;
            if(pre != -1){
                ans = max(ans,bucket[i].first - bucket[pre].second);
            }
            pre = i;
        }
        return ans;
    }
};
/*
class Solution {
public:
    int maximumGap(vector<int>& nums) {
        int n = nums.size();
        if (n < 2) {
            return 0;
        }
        int exp = 1;
        vector<int> buf(n);
        int maxVal = *max_element(nums.begin(), nums.end());
        
        while (maxVal >= exp) {
            vector<int> cnt(10);
            for (int i = 0; i < n; i++) {
                int digit = (nums[i] / exp) % 10;
                cnt[digit]++;
            }
            for (int i = 1; i < 10; i++) {
                cnt[i] += cnt[i - 1];
            }
            for (int i = n - 1; i >= 0; i--) {
                int digit = (nums[i] / exp) % 10;
                buf[cnt[digit] - 1] = nums[i];
                cnt[digit]--;
            }
            copy(buf.begin(), buf.end(), nums.begin());
            exp *= 10;
        }
        
        int ret = 0;
        for (int i = 1; i < n; i++) {
            ret = max(ret, nums[i] - nums[i - 1]);
        }
        return ret;
    }
};
*/
int main()
{
    vector<int> nums{3,6,9,1};
    Solution s;
    cout<<s.maximumGap(nums);
    return 0;
}

今天也是爱zz的一天哦！