【leetcode刷题】T202-最少移动次数使数组元素相等 II

木又连续日更第68天(68/100)html



木又的第202篇leetcode解题报告python

数学类型第18篇解题报告c++

leetcode第462题:最少移动次数使数组元素相等 IIweb

https://leetcode-cn.com/problems/minimum-moves-to-equal-array-elements-ii/数组



【题目】微信

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

例如:
输入:
[1,2,3]
输出:
2
说明:
只有两个动做是必要的(记得每一步仅可以使其中一个元素加1或减1): 
[1,2,3]  =>  [2,2,3]  =>  [2,2,2]

【思路】ui

能想出这道题数学解法的人很牛逼!spa

很明显,咱们应该找数组中间的某个数,而后计算移动步骤。.net

咱们首先想到是平均数,不过好像不对,平均数极可能不是整数,须要处理;另外,好像结果并非最小值。

其实,咱们就是求解:min(abs(nums[0] - x) + abs(nums[1] - x) +… + abs(nums[n-1] - x))

那么,x应该是中位数。(我也不知道怎么解出来的【笑哭】)

咱们能够从中位数推出最小值:

假设数组元素个数是奇数2n+1,那么大于中位数的有n个,这些数移动到中位数须要x步;小于中位数的有n个,这些数移动到中位数须要y步。那么总的移动步数为x+y步。若是不是中位数,好比中位数-1,那么大于中位数的,须要移动x+n步,小于中位数的,至少须要移动y-n步(有的数可能大于中位数-1),总的步数大于x+y步

假设数组元素个数是偶数,只要是两个中位数之间的数,均可以做为移动后的值。由于,若是小于中位数m1的数所有移动到m1,大于中位数m2(m2>=m1)的数所有移动到买m2,那么只要目标值在m1->m2之间,总的移动步数是肯定的。

【代码】

python版本

class Solution(object):
    def minMoves2(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """

        nums.sort()
        target = nums[len(nums) // 2]
        return sum(map(lambda x: abs(x - target), nums))

C++版本

class Solution {
public:
    int minMoves2(vector<int>& nums) {
        sort(nums.begin(), nums.end());
        int target = nums[nums.size() / 2];
        int sum = 0;
        for (auto n : nums){
            sum += abs(n - target);
        }
        return sum;
    }
};


前一篇文章:T201-最小移动次数使数组元素相等


给我好看


本文分享自微信公众号 - 木又AI帮(gh_eaa31cab4b91)。
若有侵权,请联系 support@oschina.cn 删除。
本文参与“OSC源创计划”,欢迎正在阅读的你也加入,一块儿分享。

相关文章
相关标签/搜索