leetcode1552. 两球之间的磁力

题目描述:

https://leetcode-cn.com/problems/magnetic-force-between-two-balls/

输入position为一个list,表示有篮子的位置,m表示小球的个数,需要找到m个小球间每两个小球的最小距离,并使其最大化。

参考:https://leetcode-cn.com/problems/magnetic-force-between-two-balls/solution/ru-men-ji-python_bao-zheng-kan-dong-by-icarus-9/

解法:

思路:采用二分法

<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/

 

 

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值