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

版权声明:本文为m0_52000372原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。