LeetCode每日一题(25)——最少移动次数使数组元素相等 II

简介: 最少移动次数使数组元素相等 II1.题目2.示例3.思路4.代码

1.题目


给定整数数组 nums 和整数 k,请返回数组中第 k 个最大的元素。


请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。


2.示例


示例 1:


输入:nums = [1,2,3]

输出:2

解释: 只需要两步操作(每步操作指南使一个元素加 1 或减 1): [1,2,3] => [2,2,3] => [2,2,2]


示例 2:


输入:nums = [1,10,2,9]

输出:16


提示:


n == nums.length

1 <= nums.length <= 10^5

-10^9 <= nums[i] <= 10^9


3.思路


一开始我以为都变成平均数变化的次数最少,后来发现不对,变成中位数才能使变化的次数最少。只要知道是向中位数变化就没什么难的了。排序的时候寻找中位数需要用到排序,我用的是官方的大顶堆排序代码,期末考试提前不想复习这个排序了。对于大顶堆排序可以看这个题解的图示

力扣大顶堆排序寻找第k大的数


4.代码


func minMoves2(nums []int) int {
  addr := len(nums)/2+1
  //数组中的第addr个最大数,中位数
  mid := findKthLargest(nums,addr)
  ans := 0
  //遍历数组,得到每个数变成中位数需要的次数
  for _,v := range nums{
    if v>mid {
      ans = ans + v - mid
    }else {
       ans = ans + mid -v
    }
  }
  return ans
}
//大顶堆法寻找数组中第k大的数
func findKthLargest(nums []int, k int) int {
  heapSize := len(nums)
  buildMaxHeap(nums, heapSize)
  for i := len(nums) - 1; i >= len(nums) - k + 1; i-- {
    nums[0], nums[i] = nums[i], nums[0]
    heapSize--
    maxHeapify(nums, 0, heapSize)
  }
  return nums[0]
}
func buildMaxHeap(a []int, heapSize int) {
  for i := heapSize/2; i >= 0; i-- {
    maxHeapify(a, i, heapSize)
  }
}
func maxHeapify(a []int, i, heapSize int) {
  l, r, largest := i * 2 + 1, i * 2 + 2, i
  if l < heapSize && a[l] > a[largest] {
    largest = l
  }
  if r < heapSize && a[r] > a[largest] {
    largest = r
  }
  if largest != i {
    a[i], a[largest] = a[largest], a[i]
    maxHeapify(a, largest, heapSize)
  }
}
相关文章
|
3天前
|
算法
LeetCode第53题最大子数组和
LeetCode第53题"最大子数组和"的解题方法,利用动态规划思想,通过一次遍历数组,维护到当前元素为止的最大子数组和,有效避免了复杂度更高的暴力解法。
LeetCode第53题最大子数组和
|
4天前
|
存储 Java API
LeetCode------合并两个有序数组(4)【数组】
这篇文章介绍了LeetCode上的"合并两个有序数组"问题,并提供了三种解法:第一种是使用Java的Arrays.sort()方法直接对合并后的数组进行排序;第二种是使用辅助数组和双指针技术进行合并;第三种则是从后向前的双指针方法,避免了使用额外的辅助数组。
LeetCode------合并两个有序数组(4)【数组】
|
12天前
|
搜索推荐 索引 Python
【Leetcode刷题Python】牛客. 数组中未出现的最小正整数
本文介绍了牛客网题目"数组中未出现的最小正整数"的解法,提供了一种满足O(n)时间复杂度和O(1)空间复杂度要求的原地排序算法,并给出了Python实现代码。
39 2
|
13天前
|
Python
【Leetcode刷题Python】53. 最大子数组和
LeetCode第53题"最大子数组和"的Python解决方案,利用动态规划的思想,通过一次遍历数组并维护当前最大和以及全局最大和来求解。
35 2
|
13天前
|
Python
【Leetcode刷题Python】剑指 Offer 03. 数组中重复的数字
解决剑指Offer题目 "数组中重复的数字" 的Python实现方法,通过使用字典来记录数组中每个数字的出现次数,快速找出重复的数字。
22 1
LeetCode------找到所有数组中消失的数字(6)【数组】
这篇文章介绍了LeetCode上的"找到所有数组中消失的数字"问题,提供了一种解法,通过两次遍历来找出所有未在数组中出现的数字:第一次遍历将数组中的每个数字对应位置的值增加数组长度,第二次遍历找出所有未被增加的数字,即缺失的数字。
|
4天前
|
前端开发
LeetCode------移动零(5)【数组】
这篇文章介绍了LeetCode上的"移动零"问题,提出了一种使用双指针的原地操作解法,该方法首先将非零元素移动到数组前端并保持相对顺序,然后填充后续位置为零,以达到题目要求。
|
3天前
|
存储 算法
LeetCode第83题删除排序链表中的重复元素
文章介绍了LeetCode第83题"删除排序链表中的重复元素"的解法,使用双指针技术在原链表上原地删除重复元素,提供了一种时间和空间效率都较高的解决方案。
LeetCode第83题删除排序链表中的重复元素
|
3天前
|
算法
LeetCode第81题搜索旋转排序数组 II
文章讲解了LeetCode第81题"搜索旋转排序数组 II"的解法,通过二分查找算法并加入去重逻辑来解决在旋转且含有重复元素的数组中搜索特定值的问题。
LeetCode第81题搜索旋转排序数组 II
|
3天前
|
算法 索引
LeetCode第34题在排序数组中查找元素的第一个和最后一个位置
这篇文章介绍了LeetCode第34题"在排序数组中查找元素的第一个和最后一个位置"的解题方法,通过使用双指针法从数组两端向中间同时查找目标值,有效地找到了目标值的首次和最后一次出现的索引位置。
LeetCode第34题在排序数组中查找元素的第一个和最后一个位置