41582023-03-15 16:08:02horvathabelÓvodacpp17Hibás válasz 33/50108ms20216 KiB
#include <bits/stdc++.h>
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;
for (ll i=0; i<n;i++){
	ll x;
	cin>>x;
	x--;
	ak.push_back(x);
}
for (int i=0; i<n;i++){
	ll x;
	cin>>x;
	sir.push_back(x);
}
vector<priority_queue<pair<ll,ll>, vector<pair<ll,ll>>, greater<pair<ll,ll>>>> q(m);
for (int i=0; i<n;i++){
	q[ak[i]].push({sir[i],i});
}
ll ans=0;
queue<ll> 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ÖsszpontTesztVerdiktIdőMemória
base33/50
1Elfogadva0/03ms1680 KiB
2Elfogadva0/07ms2372 KiB
3Elfogadva2/23ms2064 KiB
4Elfogadva2/23ms2312 KiB
5Elfogadva2/23ms2480 KiB
6Elfogadva2/23ms2604 KiB
7Hibás válasz0/23ms2844 KiB
8Elfogadva2/23ms2940 KiB
9Elfogadva2/23ms3156 KiB
10Elfogadva2/23ms3372 KiB
11Elfogadva2/23ms3584 KiB
12Elfogadva2/23ms3652 KiB
13Hibás válasz0/23ms3708 KiB
14Elfogadva3/33ms3752 KiB
15Hibás válasz0/314ms4992 KiB
16Hibás válasz0/328ms7712 KiB
17Elfogadva3/337ms7372 KiB
18Hibás válasz0/361ms12316 KiB
19Elfogadva3/368ms11064 KiB
20Elfogadva3/372ms13704 KiB
21Elfogadva3/387ms13932 KiB
22Hibás válasz0/4108ms20216 KiB