Leetcode 462. 最少移动次数使数组元素相等 II [快速选择]

最新推荐文章于 2022-09-10 10:59:24 发布

9ack!?

最新推荐文章于 2022-09-10 10:59:24 发布

阅读量178

点赞数 1

分类专栏：题解文章标签： leetcode 排序算法算法

本文链接：https://blog.csdn.net/apple_52296856/article/details/124871782

版权

题解专栏收录该内容

39 篇文章 0 订阅

订阅专栏

题目

思路

通过这个题才了解到快速选择算法这个东西，快速选择算法利用了快速排序的思想，求解TopK问题的时间复杂度是O(N)，C++库里面也提供了相应的函数。
这个题就是先找出nums数组里面的中位数，然后让所有其他数字都变为中位数。

代码

class Solution {
    public:
        int minMoves2(vector<int>& nums) {
            nth_element(nums.begin(), nums.begin()+nums.size()/2, nums.end());
            int target = nums[nums.size()/2];
            int ans = 0;
            for(int x : nums) ans += abs(x-target);
            return ans;
        }
};