题目
给你一个长度为 n 的整数数组 nums ,返回使所有数组元素相等需要的最少移动数。
在一步操作中,你可以使数组中的一个元素加 1 或者减 1 。
示例
示例 1:
输入:nums = [1,2,3]
输出:2
解释: 只需要两步操作(每步操作指南使一个元素加 1 或减 1): [1,2,3] => > [2,2,3] => [2,2,2]
示例 2:
输入:nums = [1,10,2,9]
输出:16
思路
先将数组排序,然后再取排序后的中位数作为相等的那个数,设数组长度为n
- 数组长度n为奇数直接取n 对2整除得到的下标的值
- 数组长度为偶数,取n对2整除的下标的值以及前一位的值的平均值(向下取整)
题解
def minMoves2(self, nums: List[int]) -> int: if len(nums) == 1: return 0 nums = sorted(nums) # 取中位数 mid = nums[len(nums) // 2] if len(nums) % 2 != 0 else (nums[len(nums) // 2] + nums[len(nums) // 2 - 1]) // 2 res = 0 # 计算所有数与中位数的差距,即需要移动的次数 for i in range(len(nums)): # 注意 差距需要取绝对值 res += abs(mid - nums[i]) return res