力扣小白刷题之462题最少移动次数使数组元素相等II

题目描述

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

思路

参考自:https://github.com/CyC2018/CS-Notes/blob/master/notes/Leetcode%20题解%20-%20数学.md#1-%E6%94%B9%E5%8F%98%E6%95%B0%E7%BB%84%E5%85%83%E7%B4%A0%E4%BD%BF%E6%89%80%E6%9C%89%E7%9A%84%E6%95%B0%E7%BB%84%E5%85%83%E7%B4%A0%E9%83%BD%E7%9B%B8%E7%AD%89

相遇问题:移动距离最小的方式是所有元素都移动到中位数。理由如下:

  • 设 m 为中位数。 a 和 b 是 m 两边的两个元素,且 b > a。要使 a 和 b 相等,它们总共移动的次数为 b - a,这个值等于 (b - m) + (m - a),也就是把这两个数移动到中位数的移动次数之和。
  • 设数组长度为N,则可以找到 N / 2对 a 和 b 的组合,使他们都移动到 m 的位置。

解法1

先排序,时间复杂度:O(N logN)
力扣小白刷题之462题最少移动次数使数组元素相等II_第1张图片

解法2

使用快速选择找到中位数,时间复杂度O(N)

你可能感兴趣的:(leetcode)