题目描述
给定一个非空整数数组,找到使所有数组元素相等所需的最小移动数,其中每次移动可将选定的一个元素加 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)
解法2
使用快速选择找到中位数,时间复杂度O(N)