4430 2023. 03. 27 20:08:30 horvathabel Óvoda cpp17 Hibás válasz 33/50 145ms 28572 KiB
#include <bits/stdc++.h>
#include <queue>
using namespace std;
using ll=long long; 
int main() {
ll n,m;
cin>>n>>m;
vector<ll> db; 
for (int i=0; i<m;i++){
	int z;
	cin>>z;
	db.push_back(z);
}
vector<ll> ak,sir;
vector<int> dbz(m+1);
for (ll i=0; i<n;i++){
	ll x;
	cin>>x;
	x--;
	dbz[x]++;
	ak.push_back(x);
}
for (int i=0; i<n;i++){
	ll x;
	cin>>x;
	sir.push_back(x);
}
priority_queue<pair<ll,pair<ll,ll>>, vector<pair<ll,pair<ll,ll>>>, greater<pair<ll,pair<ll,ll>>>> osszgy; 
vector<priority_queue<pair<ll,ll>, vector<pair<ll,ll>>, greater<pair<ll,ll>>>> q(m);
for (int i=0; i<n;i++){
	osszgy.push({sir[i],{ak[i],i}});
}
ll ans=0;
queue<ll> plussz;
int cnt=0; 
for (int i=0; i<n;i++) q[ak[i]].push({sir[i],i});


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();
		dbz[i]--;
	}

	
}
for (int i=0; i<m;i++){
	if (q[i].size()==0 ){
		if (!plussz.empty()){
			
		ak[plussz.front()]=i;
		q[i].push({plussz.front(),0});
		plussz.pop();	
		dbz[i]++; 
	}
		
	}
}
for (int i=0; i<m;i++){
	if (q[i].size()==0){
			while (dbz[osszgy.top().second.first]==1) osszgy.pop();
			ans+=osszgy.top().first; 
			q[i].push({0,0});
			ak[osszgy.top().second.second]=i;
			osszgy.pop();
}
}
cout<<ans<<endl;

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 7ms 2932 KiB
3 Elfogadva 2/2 3ms 2300 KiB
4 Elfogadva 2/2 3ms 2272 KiB
5 Elfogadva 2/2 3ms 2460 KiB
6 Elfogadva 2/2 3ms 2676 KiB
7 Hibás válasz 0/2 3ms 2892 KiB
8 Elfogadva 2/2 2ms 2972 KiB
9 Elfogadva 2/2 3ms 3096 KiB
10 Elfogadva 2/2 3ms 3188 KiB
11 Elfogadva 2/2 3ms 3312 KiB
12 Elfogadva 2/2 3ms 3532 KiB
13 Hibás válasz 0/2 3ms 3988 KiB
14 Elfogadva 3/3 3ms 3992 KiB
15 Hibás válasz 0/3 14ms 6008 KiB
16 Hibás válasz 0/3 32ms 9596 KiB
17 Elfogadva 3/3 39ms 10360 KiB
18 Hibás válasz 0/3 72ms 16828 KiB
19 Elfogadva 3/3 72ms 15720 KiB
20 Elfogadva 3/3 81ms 19112 KiB
21 Elfogadva 3/3 93ms 19364 KiB
22 Hibás válasz 0/4 145ms 28572 KiB