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)
  }
}
相关文章
|
6天前
【力扣】-- 移除链表元素
【力扣】-- 移除链表元素
16 1
|
12天前
【LeetCode 27】347.前k个高频元素
【LeetCode 27】347.前k个高频元素
23 0
|
18天前
|
算法
Leetcode 初级算法 --- 数组篇
Leetcode 初级算法 --- 数组篇
36 0
|
2月前
|
算法
LeetCode第53题最大子数组和
LeetCode第53题"最大子数组和"的解题方法,利用动态规划思想,通过一次遍历数组,维护到当前元素为止的最大子数组和,有效避免了复杂度更高的暴力解法。
LeetCode第53题最大子数组和
|
2月前
|
存储 Java API
LeetCode------合并两个有序数组(4)【数组】
这篇文章介绍了LeetCode上的"合并两个有序数组"问题,并提供了三种解法:第一种是使用Java的Arrays.sort()方法直接对合并后的数组进行排序;第二种是使用辅助数组和双指针技术进行合并;第三种则是从后向前的双指针方法,避免了使用额外的辅助数组。
LeetCode------合并两个有序数组(4)【数组】
LeetCode------找到所有数组中消失的数字(6)【数组】
这篇文章介绍了LeetCode上的"找到所有数组中消失的数字"问题,提供了一种解法,通过两次遍历来找出所有未在数组中出现的数字:第一次遍历将数组中的每个数字对应位置的值增加数组长度,第二次遍历找出所有未被增加的数字,即缺失的数字。
|
2月前
|
前端开发
LeetCode------移动零(5)【数组】
这篇文章介绍了LeetCode上的"移动零"问题,提出了一种使用双指针的原地操作解法,该方法首先将非零元素移动到数组前端并保持相对顺序,然后填充后续位置为零,以达到题目要求。
|
13天前
【LeetCode-每日一题】 删除排序数组中的重复项
【LeetCode-每日一题】 删除排序数组中的重复项
16 4
|
13天前
|
索引
Leetcode第三十三题(搜索旋转排序数组)
这篇文章介绍了解决LeetCode第33题“搜索旋转排序数组”的方法,该问题要求在旋转过的升序数组中找到给定目标值的索引,如果存在则返回索引,否则返回-1,文章提供了一个时间复杂度为O(logn)的二分搜索算法实现。
12 0
Leetcode第三十三题(搜索旋转排序数组)
|
13天前
|
算法 C++
Leetcode第53题(最大子数组和)
这篇文章介绍了LeetCode第53题“最大子数组和”的动态规划解法,提供了详细的状态转移方程和C++代码实现,并讨论了其他算法如贪心、分治、改进动态规划和分块累计法。
37 0