CodeForces 994B Knights of a Polygonal Table（STL、贪心）

http://codeforces.com/problemset/problem/994/B

 

题意：

给出n和m，有n个骑士，每个骑士的战力为ai，这个骑士有bi的钱，如果一个骑士的战力比另一个骑士的战力高，那么，他就可以夺取这个骑士的钱，但是每个骑士最多夺取m个人，问，这些骑士最多可以获得

多少钱。

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <string>
#include <math.h>
#include <algorithm>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <sstream>
const int INF=0x3f3f3f3f;
typedef long long LL;
const int mod=1e9+7;
//const double PI=acos(-1);
#define Bug cout<<"---------------------"<<endl
const int maxn=1e5+10;
using namespace std;

struct node
{
    int val;
    int num;
    LL ans;
    int pos;
}PE[maxn];

bool cmp1(node a,node b)
{
    return a.val<b.val;
}

bool cmp2(node a,node b)
{
    return a.pos<b.pos;
}

int main()
{
    int n,m;
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&PE[i].val);
        PE[i].pos=i;
    }
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&PE[i].num);
        PE[i].ans=0;
    }
    sort(PE+1,PE+1+n,cmp1);//按战力排序 
    priority_queue<int,vector<int> ,greater<int> > qe;//存放m个大的硬币数 
    LL sum=0;
    for(int i=1;i<=n;i++)
    {
        if(qe.size()<m)
        {
            qe.push(PE[i].num);
            sum+=PE[i].num;
        }
        else if(!qe.empty())
        {
            int t=qe.top();
            if(t<PE[i].num)
            {
                qe.pop();
                sum=sum-t+PE[i].num;
                qe.push(PE[i].num);
            }
        }
        PE[i+1].ans+=sum;//注意是i+1 
    }
    sort(PE+1,PE+1+n,cmp2);//按位置排序 
    for(int i=1;i<=n;i++)
    {
        printf("%lld ",PE[i].ans+PE[i].num);//注意是LL 
    }
    return 0;
}