[路飞]_leetcode-462-最少移动次数使数组元素相等 II

2,161 阅读2分钟

「这是我参与2022首次更文挑战的第40天,活动详情查看:2022首次更文挑战

[题目地址]

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

例如:

输入:
[1,2,3]

输出:
2

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

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

解题思路

首先我们思考一个问题,假如在村东头有一个超市,村西头有一个超市,村子中间有一个超市,现在要建立一个仓库,使得三个超市到仓库的总距离最小,仓库应该建立在哪里呢?
答案是村子中间。
其实本题和上述的例子是同样的道理,要想使得所有的元素相等,并且移动次数最小,就应该让所有的元素往中间移动,这样越靠近中间的元素的移动次数就会越少。
所以我们只需要获取输入数组的中位数,然后分别计算每个元素移动到中位数的步数,它们的和值就是总的最少移动次数。

代码实现

// 想要移动的次数最少,就应该所有的数字都往中间移动
// 所以移动的目标值应该是数组中所有数字的中位数
var minMoves2 = function (nums) {
  // 获取中位数
  function getMid(arr) {
    // 对输入数组排序
    arr.sort((a, b) => a - b)
    // 获取输入数组长度
    const len = nums.length
    // 如果长度为奇数,则直接返回中间的元素
    if (len % 2) return arr[len >> 1]
    // 否则需要获取数组中元素之和
    let sum = 0
    for (let i = 0; i < len; i++) sum += arr[i]
    // 然后获取在中间的两个数字
    const num1 = arr[len >> 1]
    num2 = arr[(len >> 1) - 1]
    // 返回两个数字中最靠近元素之和的那一个
    if (Math.abs(num1 - sum / 2) < Math.abs(num2 - sum / 2)) return num1
    return num2
  }
  const mid = getMid(nums)
  // 初始化结果值为 0
  let ret = 0
  // 遍历输入数组,计算与中位数的差值=> 就是每个元素移动到中位数的步数
  for (let i = 0; i < nums.length; i++) {
    ret += Math.abs(nums[i] - mid)
  }
  // 返回结果值
  return ret
}

至此我们就完成了 leetcode-462-最少移动次数使数组元素相等 II

如有任何问题或建议,欢迎留言讨论!👏🏻👏🏻👏🏻