B. Knights of a Polygonal Table multiset

这里写图片描述

题解:

  1. 将骑士按力量从小到大排序,到第i个骑士的时候,前面的i-1个骑士他都可以击败,找出金币最多的k个。
  2. 用multiset存金币最多的k个骑士的金币数,如果多余k个,则删除金币数最小的,直到只有k个数字。

代码:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int n,k;
//输入,p表示力量,c表示金币,idx表示输入时的位置 
struct node{
    ll p;ll c;int idx;  
}a[500050];

bool cmp(node x, node y){
    return x.p < y.p;
}
ll ans[500050];
int main(){
    ios::sync_with_stdio(false); 
    cin >> n >> k;
    //输入并按力量从小到大排序
    for(int i = 1;i <= n; ++i) cin >> a[i].p, a[i].idx = i;
    for(int i = 1;i <= n; ++i) cin >> a[i].c;
    sort(a+1,a+1+n,cmp);

    multiset <int> s; //存第i个骑士可以击败的不超过k个骑士的金币数 
    for(int i = 1;i <= n; ++i){
        ans[a[i].idx] = a[i].c; 
        for(auto t:s){
            ans[a[i].idx] += t;
        }
        s.insert(a[i].c);
        //如果将第i个骑士的金币数插入之后大于k个数字,就删除到只有k个 
        while(s.size() > k){
            s.erase(s.begin());
        }
    }
    for(int i = 1;i <= n; ++i) cout << ans[i] << " ";
    cout << endl;
    return 0;
}

猜你喜欢

转载自blog.csdn.net/q451792269/article/details/80728515