（简单贪心）CodeForces 994B-Knights of a Polygonal Table

朴灿烈づ我的快乐病毒、 2022-05-16 03:27 140阅读 0赞

*  # （简单贪心）CodeForces 994B-Knights of a Polygonal Table #

##  ##

--------------------

*  ## 题目链接：[B. Knights of a Polygonal Table][] ##
 *  ## 思路： ##

**题目大意是有n个杀手，每个杀手最多能杀k个Power值比他小的人，给出n个杀手各自的Power值和Money数，问每个杀手最多能获得多少Money**

定义一个结构体，内含杀手的原始索引（排序会打乱输入顺序，先记录），Power及Money，自定义对杀手Power值进行升序排序，排序后每个杀手能可能杀的人只可能位于他前面，用优先队列对前面的杀手进行入队，在k和索引范围内弹队出金钱最多的人，杀手最大金钱数等于杀的人钱数加上自身钱数

***一开始我全部用sort，会超时***，所以改用了优先队列，但优先队列在弹出金钱数最多杀手时需要同时出队，但是这些杀手在后面仍有可能被杀，所以用一个Temp优先队列先存出队的杀手，过后再入队。

*  ## 代码： ##

#include<iostream>
    #include<algorithm>
    #include<cstdio>
    #include<queue>
    using namespace std;
    #define MAX_SIZE 100005
    struct Knight{
        int index;
        long long Power;
        long long Money;
        bool operator<(Knight b)const    //不加const系统报错
        {
            return this->Money<b.Money;
        }
    };
    
    Knight Member[MAX_SIZE];
    long long Res[MAX_SIZE];   //储存杀手最大金钱
    bool Mycmp_1(Knight a,Knight b)   //对能力值升序
    {
        return a.Power<b.Power;
    }
    
    int main()
    {
        int n,k;
        while(cin>>n>>k)
        {
            priority_queue<Knight> Kill_List;
            priority_queue<Knight>Temp_que;   //暂存出队杀手
            for(int i=0;i<n;i++)
            {
                scanf("%lld",&Member[i].Power);
                Member[i].index=i;    //记录下原始索引
            }
            for(int i=0;i<n;i++)
                scanf("%lld",&Member[i].Money);
            sort(Member,Member+n,Mycmp_1);   //先对能力值升序
            Kill_List.push(Member[0]);   
            for(int i=0;i<n;i++)
            {
                Res[Member[i].index]=Member[i].Money;   //杀手自己有的钱数
                if(i==0)
                    continue;
                for(int j=0;j<k&&j<=i-1;j++)
                {
                    Temp_que.push(Kill_List.top());   //保存temp
                    Res[Member[i].index]+=Kill_List.top().Money;  //杀比自己弱的人中钱多的
                    Kill_List.pop();  //出队
                }
                while(!Temp_que.empty())
                {
                    Kill_List.push(Temp_que.top());   将temp重新回到Kill_List
                    Temp_que.pop();
                }
                Kill_List.push(Member[i]);   //已经完成的杀手入队
            }
           for(int i=0;i<n;i++)
                cout<<Res[i]<<" ";
            cout<<endl;
        }
    }

[B. Knights of a Polygonal Table]: http://codeforces.com/problemset/problem/994/B