154342025-02-19 16:04:51999Óvodacpp17Hibás válasz 0/50105ms11404 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-1;
            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/21ms508 KiB
5Hibás válasz0/21ms564 KiB
6Hibás válasz0/21ms316 KiB
7Hibás válasz0/21ms316 KiB
8Hibás válasz0/21ms500 KiB
9Hibás válasz0/21ms316 KiB
10Hibás válasz0/21ms316 KiB
11Hibás válasz0/21ms316 KiB
12Hibás válasz0/21ms316 KiB
13Hibás válasz0/22ms316 KiB
14Hibás válasz0/31ms316 KiB
15Hibás válasz0/312ms1316 KiB
16Hibás válasz0/327ms2740 KiB
17Hibás válasz0/335ms2836 KiB
18Hibás válasz0/361ms6376 KiB
19Hibás válasz0/363ms5080 KiB
20Hibás válasz0/370ms5676 KiB
21Hibás válasz0/383ms6448 KiB
22Hibás válasz0/4105ms11404 KiB