题目描述:
https://leetcode-cn.com/problems/magnetic-force-between-two-balls/
输入position为一个list,表示有篮子的位置,m表示小球的个数,需要找到m个小球间每两个小球的最小距离,并使其最大化。
解法:
思路:采用二分法
<1>先将position按从小到大排列。
<2>m个小球中任意两个小球之间的最小磁力min为position中相邻最短距离,
最大磁力max:position 中最小最大位置之间,m个小球平均摆放(只有在position连续的特殊情况才可以实现)
<3>一般情况下,两小球间的距离在以上两种极端情况之间,采用二分法对距离空间进行搜索。
mid = (min+max)/2 验证是否存在最小距离等于mid的放置方法:第一个小球放在第一个篮子,如果到下一个篮子距离比mid大,则将下一个小球放在第二个篮子,否则判断下一个篮子,以此类推,判断在position内能否放完m个小球。
如果最小距离可以取到mid,则扩大距离空间,若无法取到mid则减小距离空间。
def judge(dis,pos,m): count = 1 now = pos[0] for i in range(1,len(pos)): if pos[i]-now>=dis: count = count+1 now = pos[i] if count==m: return True return False def maxDistance(position, m): position.sort() maxdis = position[-1]-position[0] mindis = float('inf') for i in range(len(position)-1): dis = position[i+1]-position[i] if dis<mindis: mindis = dis left = mindis right = maxdis while right-left>1: mid = (left+right)//2 if judge(mid,position,m): left = mid else: right = mid return left position = [1,2,3,4,7] m = 3 result = maxDistance(position,m) print(result) 关于二分法区间设置:https://leetcode-cn.com/problems/binary-search/solution/li-qing-er-fen-cha-zhao-by-hello1100101/