231 2021. 03. 05 10:49:59 mraron Óvoda cpp14 Elfogadva 50/50 50ms 13524 KiB
#include<iostream>
#include<vector>
#include<map>
#include<set>
#include<cassert>
#include<cassert>
#include<unordered_map>
#include<unordered_set>
#include<functional>
#include<queue>
#include<stack>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<sstream>
#include<iomanip>
#include<cstdio>
#include<cstdlib>
#include<numeric>
#include<random>
#include<chrono>
#include<bitset>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;

#define all(x) (x).begin(), (x).end()
#define pb push_back
#define eb emplace_back
#define xx first
#define yy second
#define sz(x) (int)(x).size()
#define gc getchar
#define IO ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0)
#define mp make_pair

#ifndef ONLINE_JUDGE
#  define LOG(x) (cerr << #x << " = " << (x) << endl)
#else
#  define LOG(x) ((void)0)
#endif

using ll = long long;
using ull = unsigned long long ;
using ld = long double ;
using str = string;
using ordered_set=tree<pair<int,int>, null_type, less<pair<int,int>>, rb_tree_tag, tree_order_statistics_node_update>;

const double PI=acos(-1);
const ll INF = 1LL<<62;
const ll MINF = -(1LL<<62);

template<typename T> T getint() {
	T val=0;
	char c;
	
	bool neg=false;
	while((c=gc()) && !(c>='0' && c<='9')) {
		neg|=c=='-';
	}

	do {
		val=(val*10)+c-'0';
	} while((c=gc()) && (c>='0' && c<='9'));

	return val*(neg?-1:1);
}

//mt19937 rng(chrono::steady_clock::now().time_since_epoch().count()); uniform_int_distribution<int>(0, n-1)(rng)


struct ch {
	int xx,yy,ind;
	bool operator<(const ch& masik) const {return mp(xx,yy)<mp(masik.xx,masik.yy);}
};

int main() {
	IO;
	int n,k;
	cin>>n>>k;
	vector<int> lim(k), volt(k);
	for(int i=0;i<k;++i) {
		cin>>lim[i];
	}
	vector<ch> chs(n);
	for(int i=0;i<n;++i) cin>>chs[i].yy, chs[i].yy--;
	for(int i=0;i<n;++i) cin>>chs[i].xx, chs[i].ind=i;;
	
	sort(all(chs));
	reverse(all(chs));
	
	int res=0;
	vector<int> ans(n), lst, lst_act;
	int rem=n;
	for(auto i:chs) {
		if(lim[i.yy] && (rem>k || (rem==k && !volt[i.yy]))) {
			lim[i.yy]--;
			if(!volt[i.yy]) k--;
			volt[i.yy]=1;
			ans[i.ind]=i.yy;
			rem--;
		}else {
			res+=i.xx;
			lst.pb(i.ind);
		}		
	}
	
	for(int i=0;i<sz(volt);++i) if(!volt[i]) lst_act.pb(i);
	
	int ind=0;
	for(auto i:lst) {
		if(ind>=sz(lst_act))
			for(int j=0;j<sz(lim);++j) 
				if(lim[j]>0) lst_act.pb(j);
		
		ans[i]=lst_act[ind++];
		if(--lim[ans[i]]) lst_act.pb(ans[i]);
	}
	
	cout<<res<<"\n";
	for(auto i:ans) cout<<i+1<<" ";
	cout<<"\n";

	
	return 0;
	
}
Részfeladat Összpont Teszt Verdikt Idő Memória
base 50/50
1 Elfogadva 0/0 2ms 1756 KiB
2 Elfogadva 0/0 3ms 2076 KiB
3 Elfogadva 2/2 1ms 1904 KiB
4 Elfogadva 2/2 1ms 1892 KiB
5 Elfogadva 2/2 1ms 1904 KiB
6 Elfogadva 2/2 1ms 1924 KiB
7 Elfogadva 2/2 1ms 1912 KiB
8 Elfogadva 2/2 1ms 1932 KiB
9 Elfogadva 2/2 1ms 1936 KiB
10 Elfogadva 2/2 1ms 1924 KiB
11 Elfogadva 2/2 1ms 1924 KiB
12 Elfogadva 2/2 1ms 1928 KiB
13 Elfogadva 2/2 1ms 2004 KiB
14 Elfogadva 3/3 1ms 2000 KiB
15 Elfogadva 3/3 6ms 2568 KiB
16 Elfogadva 3/3 13ms 3884 KiB
17 Elfogadva 3/3 19ms 4240 KiB
18 Elfogadva 3/3 29ms 6704 KiB
19 Elfogadva 3/3 41ms 7304 KiB
20 Elfogadva 3/3 41ms 8920 KiB
21 Elfogadva 3/3 43ms 10128 KiB
22 Elfogadva 4/4 50ms 13524 KiB