好题分析---西南石油大学 I---选拔人才题解-优快云博客

这篇博客介绍了如何使用C++实现K个最大元素的问题，通过维护一个小根堆来动态调整，并在过程中计算最大子序列和。代码中涉及到了排序、堆和优先队列等数据结构的运用。

#include<bits/stdc++.h>
using namespace std;
int t;
int n,k;
typedef pair<int ,int > p;



int main()
{
    cin>>t;
    while(t--)
    {
        vector<p> a;
        cin>>n>>k;
        int b,c;
        for(int i=1;i<=n;i++)
        {
           cin>>b>>c;
           a.push_back({c,b});
        }
        sort(a.begin(),a.end());
        reverse(a.begin(),a.end());
        priority_queue<int ,vector<int>,greater<int>>q;//维护小根堆
        int sum=0,ans=0;
        for(int i=0;i<a.size();i++)
        {
           if(q.size()<k)
           {
               sum+=a[i].second;
               q.push(a[i].second);
           }
           else if(a[i].second>q.top())
           {
               sum-=q.top();
               q.pop();
               sum+=a[i].second;
               q.push(a[i].second);
            }
           
            ans=max(ans,sum*a[i].first);
       
        }
    cout<<ans<<endl;
    }
    
    
    
    
    return 0;
}

主要妙点在于优先队列的使用，首先注意题目中不超过k，所以计算的时候即使还未到达k的时候的ans也需要计算。接着来说关键思路，就是对代码可读性进行排序，进行for循环，然后维护一个优先队列，当队列不是满的时候就一直push，当队列满的时候判断当前循环的数是否比队列1最小的数来的小，是的话就进行替换并更新答案。