154352025-02-19 16:05:23999Óvodacpp17Hibás válasz 0/50112ms11436 KiB
// Source: https://usaco.guide/general/io

#include <bits/stdc++.h>
using namespace std;

#define int long long

const int inf=1e9;

int power(int a,int b){
    if(b==0)return 1;
    return power(a,b-1)*a;
}

signed main() {
    int n,k;cin>>n>>k;
    vector<int> db(k);
    for(int i = 0;i<k;i++)cin>>db[i];
    vector<int> akar(n);
    for(int i = 0;i<n;i++)cin>>akar[i];
    vector<int> cost(n);
    for(int i = 0;i<n;i++)cin>>cost[i];



    vector<vector<pair<int,int>>> full(k);
    for(int i = 0;i<n;i++){
        akar[i]--;
        full[akar[i]].push_back({cost[i],i});
    }
    for(int i = 0;i<k;i++){
        sort(full[i].rbegin(),full[i].rend());
    }/*// debug kiiras
    for(int i = 0;i<k;i++){
        for(auto j:full[i])cout<<j.second<<' ';
        cout<<endl;
    }*/
    vector<int> ans(n,-1),dbb;
    dbb=db;
    for(int i = 0;i<k;i++){
        for(int j = 0;j<min(db[i],(int)full[i].size());j++){
            ans[full[i][j].second]=i;
            dbb[i]--;
        }
    }
    db=dbb;
    stack<pair<int,int>> s;
    for(int i = 0;i<k;i++){
        if(db[i]>0){
            s.push({i,db[i]});
            //cerr<<i<<' '<<db[i]<<endl;
        }
    }
    int cryingsimulatortime=0;
    for(int i = 0;i<n;i++){
        if(ans[i]==-1){
            cryingsimulatortime+=cost[i];// >:)
            ans[i]=s.top().first;
            s.top().second--;
            if(s.top().second==0)s.pop();
        }
    }
    for(int i = 0;i<n;i++)cout<<ans[i]+1<<' ';
}
RészfeladatÖsszpontTesztVerdiktIdőMemória
base0/50
1Hibás válasz0/01ms316 KiB
2Hibás válasz0/04ms564 KiB
3Hibás válasz0/21ms316 KiB
4Hibás válasz0/21ms316 KiB
5Hibás válasz0/21ms316 KiB
6Hibás válasz0/21ms316 KiB
7Hibás válasz0/21ms316 KiB
8Hibás válasz0/21ms384 KiB
9Hibás válasz0/21ms316 KiB
10Hibás válasz0/21ms316 KiB
11Hibás válasz0/21ms316 KiB
12Hibás válasz0/21ms320 KiB
13Hibás válasz0/22ms568 KiB
14Hibás válasz0/32ms316 KiB
15Hibás válasz0/312ms1408 KiB
16Hibás válasz0/328ms2800 KiB
17Hibás válasz0/335ms2944 KiB
18Hibás válasz0/364ms6376 KiB
19Hibás válasz0/364ms5184 KiB
20Hibás válasz0/374ms5536 KiB
21Hibás válasz0/386ms6452 KiB
22Hibás válasz0/4112ms11436 KiB