跳转至

LeetCode: 462. 最少移动次数使数组元素相等 II

1、题目描述

给定一个非空整数数组,找到使所有数组元素相等所需的最小移动数,其中每次移动可将选定的一个元素加1或减1。 您可以假设数组的长度最多为10000。

例如:

输入:
[1,2,3]

输出:
2

说明:
只有两个动作是必要的(记得每一步仅可使其中一个元素加1或减1): 

[1,2,3]  =>  [2,2,3]  =>  [2,2,2]

2、解题思路

这道题每一次仅移动一个元素,为了保证总的移动次数最少,我们就需要找到中位数,这样移动次数是最少的
例如
[1 4 5]
中位数是多少呢,是4,因此,移动次数就是3+1 = 4
class Solution:
    def minMoves2(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        median = sorted(nums)[len(nums) // 2]
        return sum([abs(median - i) for i in nums])