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大佬的二分查找算法模板
内容来源于网络如有侵权请私信删除
文章来源: 博客园
- 还没有人评论，欢迎说说您的想法！
客服
