6155. 找出数组的第 K 大和 --- 堆 + 二进制思维

最新推荐文章于 2024-07-25 11:15:54 发布

_WAWA鱼_

最新推荐文章于 2024-07-25 11:15:54 发布

阅读量98

点赞数

分类专栏：比赛补题文章标签： leetcode 算法动态规划

本文链接：https://blog.csdn.net/qq_51282224/article/details/126455521

版权

比赛补题专栏收录该内容

99 篇文章 5 订阅

订阅专栏

题目链接

class Solution {
public:
    long long kSum(vector<int> &nums, int k) {
        long long sum = 0;
        for(auto &x : nums)
        {
            if(x >= 0) sum += x;
            else x = -x;
        }
        sort(nums.begin(), nums.end());

        priority_queue<pair<long long, int>> heap;
        heap.push({sum, 0});
        while(-- k)
        {
            auto [s, i] = heap.top();
            heap.pop();
            if(i < nums.size())
            {
                heap.push({s - nums[i], i + 1});
                if(i) heap.push({s - nums[i] + nums[i - 1], i + 1});
            }
        }
        return heap.top().first;
    }
};