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