leetcode1552题解【二分+贪心】
作者:互联网
leetcode1552.两球之间的磁力
算法
二分+贪心
时间复杂度O(nlogn + nlogm)
1.根据题意描述,我们需要将m个球放入到n个篮子中,根据题目中数据范围描述发现m <= n,故可以将一个球放入到一个篮子中。这道题主要就是要求出相邻的两个球之间的距离的最小值,而且要尽可能的让这个最小值最大化
2.分析完了题意,下面来分析一下如何解题。刚开始的思路是首先排序,然后将第一个球放到数组的第一个位置,然后根据剩余的球的个数枚举球的位置。但因为还需要记录相邻两个球的距离差,如果这样纯暴力的话写起来太过于繁琐,并且耗时大,最终该思路未果。在搜索了大佬们的解题思路后,了解到可以使用二分思想来对磁力进行二分,具体思路如下。
3.对于一些球,它们之间磁力的最大值是poisiton数组中元素的最大值减去最小值(这里先不考虑球的个数),那么我们就可以确定了磁力的区间范围[l,r]
,然后将[l,r]
划分为[l,mid-1]
、[mid,r]
,为什么要这样划分呢,因为题目中说了要最大化最小磁力,所以我们要尽可能的使得mid更大。划分条件就是判断position数组是否能够找到m个篮子使得它能够满足相邻的两个球之间的距离大于等于mid,如果满足,则l=mid
,否则r=mid-1
。
4.为了使得最小磁力最大化,我们可以使其中一个球位于最左边的那个篮子里,然后再以此枚举球的位置,使得相邻的两个球的距离大于等于mid。
5.由此,这道题的总体思路是:先排序,然后二分。
C++代码
class Solution {
public:
int maxDistance(vector<int>& position, int m) {
sort(position.begin(), position.end());
int len = position.size();
int l = 1, r = position[len - 1] - position[0];
while(l < r){
int mid = l + r + 1>> 1;
if(check(mid, position, m))
l = mid;
else
r = mid - 1;
}
return l;
}
bool check(int k, vector<int>& position, int m){
int len = position.size();
int last = position[0]; //last用于记录上一个存放球的篮子的位置
int t = 1; //记录已经放入到篮子的球的个数
for(int i = 1; i < len; i++){
if(position[i] - last >= k){
++t;
last = position[i];
if(t == m) return true;
}
}
return false;
}
};
这道题使用的二分模板来源于yxc大佬的二分查找算法模板
标签:二分,leetcode1552,int,题解,mid,len,position,磁力,贪心 来源: https://www.cnblogs.com/KeepZ/p/13721495.html