1339 | 2022. 05. 14 19:21:49 | nkdorka1212 | Óvoda | cpp11 | Hibás válasz 33/50 | 82ms | 28572 KiB |
#include <bits/stdc++.h>
using namespace std;
#pragma GCC target ("avx2")
#pragma GCC optimization ("O3")
#pragma GCC optimization ("unroll-loops")
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n,k;
cin>>n>>k;
vector<int>szerep(k+1);
vector<int>marad(k+1);
vector<int>v1(n+1);
vector<int>v2(n+1);
vector<vector<pair<int,int>>>g(n+1);
vector<int>veg;
vector<int>mo(n+1);
for(int i=1;i<=k;i++)
{
cin>>szerep[i];
}
marad=szerep;
for(int i=1;i<=n;i++)
{
cin>>v1[i];
if(marad[v1[i]]>0)
{
marad[v1[i]]--;
}
}
for(int i=1;i<=n;i++)
{
cin>>v2[i];
}
for(int i=1;i<=n;i++)
{
g[v1[i]].push_back({v2[i],i});
}
for(int i=1;i<=k;i++)
{
sort(g[i].rbegin(),g[i].rend());
}
for(int i=1;i<=k;i++)
{
if(g[i].size()==0)
{
veg.push_back(i);
marad[i]--;
}
}
int cnt=0;
for(int i=1;i<=k;i++)
{
while(marad[i]>0)
{
marad[i]--;
veg.push_back(i);
}
}
long long sir=0;
vector<bool>ok(k+1,0);
for(int i=1;i<=k;i++)
{
for(auto x:g[i])
{
if(szerep[i]>0)
{
szerep[i]--;
mo[x.second]=i;
ok[i]=true;
}else
{
mo[x.second]=veg[cnt];
ok[veg[cnt]]=true;
cnt++;
sir+=x.first;
}
}
}
/*for(int x:ok)
{
cout<<x<<" ";
}*/
cout<<endl;
vector<pair<int,int>>elek(n+1);
for(int i=1;i<=n;i++)
{
elek[i]={v2[i],i};
}
sort(elek.begin(),elek.end());
int h=1;
for(int i=1;i<=k;i++)
{
if(!ok[i])
{
mo[elek[h].second]=i;
sir+=elek[h].first;
}
}
cout<<sir<<'\n';
for(int i=1;i<=n;i++)
{
cout<<mo[i]<<" ";
}
return 0;
}
/*
for(int i=1;i<=k;i++)
{
cout<<i<<'\n';
for(auto x:g[i])
{
cout<<x.first<<" "<<x.second<<" "<<'\n';
}
}
6 4
2 2 2 2
1 2 4 1 2 2
1 4 2 1 2 3
6 4
2 1 4 1
2 2 1 3 1 1
4 5 8 9 10 1
*/
Részfeladat | Összpont | Teszt | Verdikt | Idő | Memória | ||
---|---|---|---|---|---|---|---|
base | 33/50 | ||||||
1 | Elfogadva | 0/0 | 2ms | 1812 KiB | |||
2 | Elfogadva | 0/0 | 4ms | 2564 KiB | |||
3 | Elfogadva | 2/2 | 1ms | 1900 KiB | |||
4 | Elfogadva | 2/2 | 1ms | 1904 KiB | |||
5 | Elfogadva | 2/2 | 1ms | 1908 KiB | |||
6 | Elfogadva | 2/2 | 1ms | 1912 KiB | |||
7 | Hibás válasz | 0/2 | 1ms | 1920 KiB | |||
8 | Elfogadva | 2/2 | 1ms | 1928 KiB | |||
9 | Elfogadva | 2/2 | 1ms | 1948 KiB | |||
10 | Elfogadva | 2/2 | 1ms | 1928 KiB | |||
11 | Elfogadva | 2/2 | 1ms | 1940 KiB | |||
12 | Elfogadva | 2/2 | 1ms | 1948 KiB | |||
13 | Hibás válasz | 0/2 | 1ms | 2028 KiB | |||
14 | Elfogadva | 3/3 | 2ms | 2040 KiB | |||
15 | Hibás válasz | 0/3 | 8ms | 3848 KiB | |||
16 | Hibás válasz | 0/3 | 18ms | 6308 KiB | |||
17 | Elfogadva | 3/3 | 28ms | 8900 KiB | |||
18 | Hibás válasz | 0/3 | 43ms | 14996 KiB | |||
19 | Elfogadva | 3/3 | 45ms | 14356 KiB | |||
20 | Elfogadva | 3/3 | 50ms | 16612 KiB | |||
21 | Elfogadva | 3/3 | 57ms | 18640 KiB | |||
22 | Hibás válasz | 0/4 | 82ms | 28572 KiB |