Educational Codeforces Round 62 (Rated for Div. 2) - C Playlist

当时题意看错了。。。不过大致思路是对的,唯一没有想到的就是用优先队列搞这个东西,真是不该啊。。。

 题意大概就是,有N首歌,N首歌有两个东西,一个是长度Ti,一个是美丽值Bi,你最多可以选择K首歌,

这些首歌的总和是这些歌的总的长度乘以其中最大的美丽值。

简而言之,就是选择最K个点,问其第一权值总和乘以最小第二权值最大时多少?

很容易想到需要把第二权值排序。

举个例子

4 3 

4 7

15 1

3 6

6 8

排序后

T 15   3      4      6

B 1     6      7      8

我们发现,如果我选择了第一个,我需要在剩下的中找到k-1个第一个权值最大的数,并求和。

同样的,我们如果选择了第二个,那么第一个一定是不能选的,我们也是需要在剩下的数中找到前k-1个第一权值最大的数,并求和。

想到这里,离成功就很近了,但是我就没想下去,就觉得已经不可能了。。。

我们不妨这样考虑,如果我逆向枚举呢???

我从N-K+1开始,那么第K个一定被选,后K-1个也一定必选对吧?

如果我从N-K开始,那么第K-1个一定被选,后K个中我们找到其中的前K-1个第一权值大的数?

如果我从N-K-1开始,那么第K-2个一定被选,后K个中我们找到其中前K-1个第一权值大的数?

关键问题在,如何每次维护前K-1个第一权值大的数,而且还要维护其和?

那么优先队列便呼之欲出了,首先维护后K项的和,每次往后选择的时候,这时优先队列有K个值,我们需要把,优先队列中,最小的给那走,并加上当前必选的第二权值,再维护和。这样优先队列里面一直维护的从从枚举点后K-1个最大值。但是需要注意的是,题意是要求最多K个,那么我们再把后k项的值再最开始一并维护就行了。

#include<iostream>
#include<string.h>
#include<algorithm>
#include<stdio.h>
#include<queue>
#define rep(i,j,k) for(int i=j;i<=k;i++)
#define per(i,j,k) for(int i=j;i>=k;i--)
#define LL long long
using namespace std;
const int maxx = 3e5+6;
struct node
{
    int t,b;
} a[maxx];
bool cmp(node x,node y)
{
    if (x.b==y.b)
    {
        return x.t<y.t;
    }
    return x.b<y.b;
}
priority_queue<int,vector<int>,greater<int> >q;
int main()
{
    int n,k;
    while(~scanf("%d%d",&n,&k))
    {

        for (int i=1; i<=n; i++)
        {
            scanf("%d%d",&a[i].t,&a[i].b);
        }
        sort(a+1,a+1+n,cmp);
        LL sum=0;
        LL ans=0;
        per(i,n,n-k+1)
        {
            q.push(a[i].t);
            sum+=(LL)a[i].t;
            ans=max(ans,sum*a[i].b);
        }
        if (k<n)
        {
            LL num;
            per(i,n-k,1)
            {
                sum-=q.top();
                sum+=(LL)a[i].t;
                q.pop();
                q.push(a[i].t);
                num=(LL)sum*a[i].b;
                ans=max(ans,num);
            }
        }
        printf("%lld\n",ans);
    }
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/bluefly-hrbust/p/10586150.html