4053 2023. 03. 11 10:00:05 1478 Óvoda cpp17 Részben helyes 40/50 65ms 20380 KiB
#include <bits/stdc++.h>
#include <random>

using namespace std;

typedef long long ll;
typedef long double ld;
typedef pair<int, int> p32;
typedef pair<ll, ll> p64;
typedef pair<double, double> pdd;
typedef vector<ll> v64;
typedef vector<int> v32;
typedef vector<vector<int>> vv32;
typedef vector<vector<ll>> vv64;
typedef vector<vector<p64>> vvp64;
typedef vector<p64> vp64;
typedef vector<p32> vp32;
ll MOD = 998244353;
double eps = 1e-12;
#define forn(i, e) for (ll i = 0; i < e; i++)
#define forsn(i, s, e) for (ll i = s; i < e; i++)
#define rforn(i, s) for (ll i = s; i >= 0; i--)
#define rforsn(i, s, e) for (ll i = s; i >= e; i--)
#define ln "\n"
#define dbg(x) cout << #x << " = " << x << ln
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define INF 2e18
#define fast_cin()                    \
    ios_base::sync_with_stdio(false); \
    cin.tie(NULL);                    \
    cout.tie(NULL)
#define all(x) (x).begin(), (x).end()
#define sz(x) ((ll)(x).size())

int main()
{
    fast_cin();
    //ifstream cin("in.txt");

    int n;
    cin >> n;

    int k;
    cin >> k;

    vector<int> cap(k + 1);
    for (int i = 1; i <= k; i++)
    {
        cin >> cap[i];
    }

    vector<int> ask(n + 1);
    for (int i = 1; i <= n; i++)
    {
        cin >> ask[i];
    }

    vector<int> price(n + 1);
    for (int i = 1; i <= n; i++)
    {
        cin >> price[i];
    }

    vector<vector<pair<int, int>>> al(n + 1);
    for (int i = 1; i <= n; i++)
    {
        al[ask[i]].push_back({price[i], i});
    }

    for (int i = 1; i <= k; i++)
    {
        sort(al[i].rbegin(), al[i].rend());
    }

    long long ans = 0;

    vector<pair<int, int>> rem;
    for (int i = 1; i <= k; i++)
    {
        while (al[i].size() > cap[i])
        {
            rem.push_back(al[i][al[i].size() - 1]);
            ans += al[i][al[i].size() - 1].first;
            al[i].pop_back();
        }
    }
    sort(rem.rbegin(), rem.rend());

    priority_queue<pair<int,int>> pq;
    for (int i = 1; i <= k; i++)
    {
        if(al[i].size() == 0){
            pq.push({INT_MAX, i});
        }
        else{
            pq.push({cap[i] - al[i].size(), i});
        }
    }

    while(rem.size()){
        int i = pq.top().second;
        al[i].push_back(rem[rem.size() - 1]);
        pq.pop();
        pq.push({cap[i] - al[i].size(), i});
        rem.pop_back();
    }

    for (int i = 1; i <= k; i++)
    {
        for (int j = al[i].size() - 1; j >= 1; j--)
        {
            rem.push_back(al[i][j]);
        }
    }
    sort(rem.rbegin(), rem.rend());

    for(int i = 1; i <= k; i++){
        if(al[i].size() == 0){
            ans += rem[rem.size() - 1].first;
            al[i].push_back(rem[rem.size() - 1]);
            rem.pop_back();
        }
    }

    vector<int> anss(n + 1);
    for(int i = 1; i <= k; i++){
        for(int j = 0; j < al[i].size(); j++){
            anss[al[i][j].second] = i;
        }
    }
    
    cout << ans << '\n';
    for(int i = 1; i <= n; i++){
        cout << anss[i] << " ";
    }

    return 0;
}
Részfeladat Összpont Teszt Verdikt Idő Memória
base 40/50
1 Elfogadva 0/0 3ms 1828 KiB
2 Elfogadva 0/0 4ms 2768 KiB
3 Elfogadva 2/2 2ms 2256 KiB
4 Elfogadva 2/2 2ms 2336 KiB
5 Elfogadva 2/2 2ms 2568 KiB
6 Elfogadva 2/2 2ms 2544 KiB
7 Elfogadva 2/2 3ms 2676 KiB
8 Elfogadva 2/2 3ms 2836 KiB
9 Elfogadva 2/2 3ms 3060 KiB
10 Elfogadva 2/2 3ms 3276 KiB
11 Elfogadva 2/2 2ms 3364 KiB
12 Elfogadva 2/2 2ms 3436 KiB
13 Részben helyes 1/2 3ms 3608 KiB
14 Elfogadva 3/3 3ms 3884 KiB
15 Részben helyes 1/3 8ms 5380 KiB
16 Részben helyes 1/3 17ms 7612 KiB
17 Elfogadva 3/3 25ms 9088 KiB
18 Részben helyes 1/3 35ms 12680 KiB
19 Elfogadva 3/3 43ms 13500 KiB
20 Elfogadva 3/3 54ms 15584 KiB
21 Elfogadva 3/3 57ms 15636 KiB
22 Részben helyes 1/4 65ms 20380 KiB