code forces 994B

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5+5;
int n,k;
struct node {
    int p,id;
} a[maxn];
bool cmp(node a,node b) {
    return a.p<b.p;
}
vector<long long> v;
long long c[maxn];
long long ans[maxn];
int main() {
    ios::sync_with_stdio(false);
    cin>>n>>k;
    for(int i=0; i<n; i++) {
        cin>>a[i].p;
        a[i].id=i;
    }
    for(int i=0; i<n; i++) {
        cin>>c[i];
    }
    //k=0时特殊处理
    if(k==0) {
        for(int i=0; i<n; i++) {
            printf("%d ",c[i]);
        }
        printf("\n");
        return 0;
    } else {
        //对力量值排序
        sort(a,a+n,cmp);
        ans[a[0].id]=0;
        long long sumk=0;//前k大和
        for(int i=1; i<n; i++) {
            if(v.size()<k) {
                /在k个数之前时  加上前面这些数的cost
                v.push_back(c[a[i-1].id]);
                sumk+=c[a[i-1].id];//前i个数的和
            } else if(c[a[i-1].id]>v[0]) {
                //前面数中最大的k个数的和
                sumk+=c[a[i-1].id]-v[0];//更新最大值
                v[0]=c[a[i-1].id];//更新第k大值
            }
            sort(v.begin(),v.end());//对于k个数重新排序
            ans[a[i].id]=sumk;
        }
        for(int i=0; i<n; i++) {
            printf("%lld ",ans[i]+c[i]);
        }
        printf("\n");

    }
    return 0;
}