4157 | 2023. 03. 15 16:00:56 | horvathabel | Óvoda | cpp17 | Hibás válasz 33/50 | 108ms | 17104 KiB |
#include <bits/stdc++.h>
using namespace std;
int main() {
int n,m;
cin>>n>>m;
vector<int> db;
for (int i=0; i<m;i++){
int z;
cin>>z;
db.push_back(z);
}
vector<int> ak,sir;
for (int i=0; i<n;i++){
int x;
cin>>x;
x--;
ak.push_back(x);
}
for (int i=0; i<n;i++){
int x;
cin>>x;
sir.push_back(x);
}
vector<priority_queue<pair<int,int>, vector<pair<int,int>>, greater<pair<int,int>>>> q(m);
for (int i=0; i<n;i++){
q[ak[i]].push({sir[i],i});
}
int ans=0;
queue<int> plussz;
for (int i=0; i<m;i++){
while (q[i].size()>db[i]){
plussz.push(q[i].top().second);
ans+=q[i].top().first;
q[i].pop();
}
}
cout<<ans<<endl;
for (int i=0; i<m;i++){
if (q[i].size()==0 && !plussz.empty()){
ak[plussz.front()]=i;
q[i].push({plussz.front(),0});
plussz.pop();
}
}
for (int i=0; i<m;i++){
while (q[i].size()<db[i] && !plussz.empty()){
ak[plussz.front()]=i;
q[i].push({plussz.front(),0});
plussz.pop();
}
}
for (auto i:ak) cout<<i+1<<" ";
}
Részfeladat | Összpont | Teszt | Verdikt | Idő | Memória | ||
---|---|---|---|---|---|---|---|
base | 33/50 | ||||||
1 | Elfogadva | 0/0 | 3ms | 1808 KiB | |||
2 | Elfogadva | 0/0 | 6ms | 2480 KiB | |||
3 | Elfogadva | 2/2 | 3ms | 2212 KiB | |||
4 | Elfogadva | 2/2 | 3ms | 2368 KiB | |||
5 | Elfogadva | 2/2 | 3ms | 2564 KiB | |||
6 | Elfogadva | 2/2 | 3ms | 2804 KiB | |||
7 | Hibás válasz | 0/2 | 3ms | 3016 KiB | |||
8 | Elfogadva | 2/2 | 3ms | 3232 KiB | |||
9 | Elfogadva | 2/2 | 3ms | 3456 KiB | |||
10 | Elfogadva | 2/2 | 3ms | 3632 KiB | |||
11 | Elfogadva | 2/2 | 3ms | 3836 KiB | |||
12 | Elfogadva | 2/2 | 3ms | 3956 KiB | |||
13 | Hibás válasz | 0/2 | 3ms | 4128 KiB | |||
14 | Elfogadva | 3/3 | 3ms | 3980 KiB | |||
15 | Hibás válasz | 0/3 | 14ms | 5100 KiB | |||
16 | Hibás válasz | 0/3 | 28ms | 7848 KiB | |||
17 | Elfogadva | 3/3 | 37ms | 6472 KiB | |||
18 | Hibás válasz | 0/3 | 61ms | 10876 KiB | |||
19 | Elfogadva | 3/3 | 65ms | 8388 KiB | |||
20 | Elfogadva | 3/3 | 71ms | 9456 KiB | |||
21 | Elfogadva | 3/3 | 86ms | 10148 KiB | |||
22 | Hibás válasz | 0/4 | 108ms | 17104 KiB |