【LeetCode】1552. 两球之间的磁力

题目

在代号为 C-137 的地球上,Rick 发现如果他将两个球放在他新发明的篮子里,它们之间会形成特殊形式的磁力。Rick 有 n 个空的篮子,第 i 个篮子的位置在 position[i] ,Morty 想把 m 个球放到这些篮子里,使得任意两球间 最小磁力 最大。

已知两个球如果分别位于 x 和 y ,那么它们之间的磁力为 |x - y| 。
给你一个整数数组 position 和一个整数 m ,请你返回最大化的最小磁力。

示例 1:

在这里插入图片描述

输入:position = [1,2,3,4,7], m = 3
输出:3
解释:将 3 个球分别放入位于 1,4 和 7 的三个篮子,两球间的磁力分别为 [3, 3, 6]。最小磁力为 3 。我们没办法让最小磁力大于 3 。

示例 2:

输入:position = [5,4,3,2,1,1000000000], m = 2
输出:999999999
解释:我们使用位于 1 和 1000000000 的篮子时最小磁力最大。

提示:

n == position.length
2 <= n <= 10^5
1 <= position[i] <= 10^9
所有 position 中的整数 互不相同 。
2 <= m <= position.length

题解

使用二分法
将position进行排序
最小间距一定是相邻小球坐标差值
使用二分法不断改变mid(间隙),遍历position,查找能够填入的小球数
如果cnt大于等于m,则增大mid来减少cnt
如果cnt小于m,则减少mid来增大cnt

这一题类似于【LeetCode】1760.袋子里最少数目的球

class Solution {
    
    
public:
    int maxDistance(vector<int>& position, int m) {
    
    
        int len = position.size();
        sort(position.begin(),position.end());
        
        int left = 1;
        int right = position[len-1];
        
        while(left+1 < right)
        {
    
    
            int mid = left + ((right-left)>>1);
            int cnt = 1;
            int pre = position[0];
            for(int& it:position)
            {
    
    
                if( it - pre >=mid)
                {
    
    
                    cnt++;
                    pre = it;
                }
            }
            //cout<<left<<"=="<<mid<<"=="<<right<<"==>"<<cnt<<"???"<<m<<endl;
            if(cnt >= m)
            {
    
    
                left = mid;
                //cout<<"增大mid,减少cnt"<<left<<"&&"<<right<<endl;
            }
            else
            {
    
    
                right = mid;
                //cout<<"减少mid,增大cnt"<<left<<"&&"<<right<<endl;
            }
            
        }
        return left;
    }
};

猜你喜欢

转载自blog.csdn.net/qq_45972928/article/details/126210781
今日推荐