题目:
给定一个非空整数数组,找到使所有数组元素相等所需的最小移动数,其中每次移动可将选定的一个元素加1或减1。 您可以假设数组的长度最多为10000。
例如:
输入:
[1,2,3]
输出:
2
说明:
只有两个动作是必要的(记得每一步仅可使其中一个元素加1或减1):
[1,2,3] => [2,2,3] => [2,2,2]
思路:
其实最小的就是中位数,假设中位数左边差值和为x,右边为y。移动就是x+y,假设是中位数左边一个,移动最好的情况是x+y+1,然后注意下python的除法返回的是float就行了
代码
class Solution:
def minMoves2(self, nums: List[int]) -> int:
nums.sort()
mid=0
lens=len(nums)
if lens%2==0:
mid=int((nums[int(lens/2)-1]+nums[int(lens/2)])/2)
else:
mid=int(nums[(lens-1)//2])
sum=0
for i in nums:
sum+=abs(i-mid)
return sum