249232026-02-16 19:36:38999Óvodacpp17Hibás válasz 28/50112ms10616 KiB
// Source: https://usaco.guide/general/io

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

signed main() {
    int n,k;cin>>n>>k;
    vector<int> hanyan(k);
    for(int i = 0;i<k;i++){
        cin>>hanyan[i];
    }
    vector<int> sir(n),fav(n),hova(n);
    for(int i = 0;i<n;i++){
        cin>>fav[i];
    }
    vector<vector<array<int,2>>> v(k);
    for(int i = 0;i<n;i++){
        cin>>sir[i];
        fav[i]--;
        v[fav[i]].push_back({sir[i],i});
    }
    for(int i = 0;i<k;i++){
        sort(v[i].rbegin(),v[i].rend());
        //for(auto j : v[i])cout<<j[1]<<' '<<j[0]<<", ";
        //cout<<endl;
    }
    int minsiras=0;
    deque<array<int,2>> extra;
    for(int i = 0;i<k;i++){
        for(int j = 0;j<v[i].size();j++){
            if(j>=hanyan[i]){
                minsiras+=v[i][j][0];
            }
            else{
                hova[v[i][j][1]]=i+1;
            }
        }
        if(hanyan[i]>v[i].size()){
            extra.push_back({hanyan[i]-v[i].size(),i+1});
        }
    }
    cout<<minsiras<<endl;
    for(int i = 0;i<n;i++){
        if(hova[i]>0)cout<<hova[i]<<' ';
        else{
            cout<<extra[0][1]<<' ';
            extra[0][0]--;
            if(extra[0][0]==0)extra.pop_front();
        }
    }
}
RészfeladatÖsszpontTesztVerdiktIdőMemória
base28/50
1Elfogadva0/02ms492 KiB
2Hibás válasz0/06ms568 KiB
3Elfogadva2/22ms316 KiB
4Részben helyes1/21ms560 KiB
5Elfogadva2/21ms316 KiB
6Elfogadva2/21ms316 KiB
7Hibás válasz0/21ms316 KiB
8Részben helyes1/21ms316 KiB
9Elfogadva2/21ms316 KiB
10Elfogadva2/21ms316 KiB
11Elfogadva2/21ms316 KiB
12Részben helyes1/21ms316 KiB
13Hibás válasz0/22ms404 KiB
14Elfogadva3/32ms316 KiB
15Hibás válasz0/312ms1248 KiB
16Hibás válasz0/328ms2488 KiB
17Elfogadva3/335ms2912 KiB
18Hibás válasz0/361ms5984 KiB
19Elfogadva3/368ms4908 KiB
20Elfogadva3/371ms5684 KiB
21Részben helyes1/386ms6468 KiB
22Hibás válasz0/4112ms10616 KiB