CodeForce[1500-2000]——1945F - Kirill and Mushrooms

codeforce刷题日记

题目大意：此时有n个蘑菇，每个蘑菇有自己的魔力值，并给你一个1-n组成的排列数p，如果你采摘k个蘑菇，那p前k-1个元素对应的蘑菇魔力值变成0，现在题目要求如何得到 采摘蘑菇数*蘑菇魔力最小值，这个值最大，并且采摘的蘑菇数最小。

思路：

实际上我们每次枚举选择k个数之后，删除的k-1个数应该是p数组前k-1位置所描述的位置的数，也就是我们选择的数实际上只能从p数组前k-1个位置之后的所有位置中选择，根据题意我们知道最大魔力=选择的数的数量*选择的数中魔力最小的那个蘑菇的魔力，我们枚举选择了k个数，实际上就是要维护可以选择的那些数中最大的k个数，那么实际上就是需要维护一个大小为k的小根堆，用于维护后面可以选择的数中最大的k个数，此时选择的数中的最小的元素就是堆顶元素，为了方便处理，我们考虑从大到小枚举要选择的元素个数i，那么堆里面的元素个数除了每次加入新的元素v[p[i]]之外，由于我们枚举的是选择i个元素，所以我们的堆中最多只能有i个元素，为了保证魔力更大，我们应该把堆中更小的多余出来的元素删除，也就是每次删除堆顶元素，直到堆中元素个数小于等于i，然后此时堆中剩余的元素就是我们要选择的i个元素，堆顶元素就是这一堆数中的最小值，然后判断一下更新答案即可。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e5+10;
int n,t;
ll v[maxn],p[maxn];
priority_queue<ll,vector<ll>,greater<ll>>q;
int main(){
    cin>>t;
    while(t--){
        while(q.size()) q.pop();
        cin>>n;
        for(int i=1;i<=n;i++)cin>>v[i];
        for(int i=1;i<=n;i++) cin>>p[i];
        ll ans=0,x=0;
        for(int i=n;i>0;i--){//这时取i个蘑菇，将[i,n]压入队，i-1到n已经在之前循环入队
            q.push(v[p[i]]);
            while(q.size()>i) q.pop();
            if(q.size()==i&&q.top()*i>=ans){ //只有[i,n]个数达到i才判断
                ans=q.top()*i;x=i;
            }
        }
        cout<<ans<<" "<<x<<endl;
    }
    return 0;
}