Codeforces 994B. Knights of a Polygonal Table

Posted 2020-11-18 zhangjiuding

解题思路

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

---

我就是因为没有用multiset在最后5分钟被hack了。

代码

#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;
}