LeetCode每日一题——462. 最少移动次数使数组元素相等 II

题目

给你一个长度为 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

1. 数组长度n为奇数直接取n 对2整除得到的下标的值
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