Submission

Score: 8

User: KantaponZ

Problemset: แคง (Kang)

Language: cpp

Time: 2.017 second

Submitted On: 2025-06-30 11:03:14

#include <bits/stdc++.h>
using namespace std;

#define ll long long

unordered_map<int,int> cnt;
unordered_map<int,bool> del;
vector<ll> C;
priority_queue<pair<ll, int>> pq;

vector<long long> capsize(vector<int> A, vector<int> B) {
      
    int N = A.size(), M = B.size();
    ll sum = 0;
    for (int i = 0; i < N; i++) {
        sum += A[i];
        cnt[A[i]]++;
    }
    for (auto [k, v] : cnt) {
        pq.push({k * v, k});
    }
    for (int i = 0; i < M; i++) {
        if (sum <= 0 || pq.empty()) {
          break;
        }
        if (!del[B[i]]) {
            sum += B[i];
            cnt[B[i]]++;
            pq.push({B[i] * cnt[B[i]], B[i]});
        }
        while (!pq.empty() && sum > 0 && del[pq.top().second] == true) {
            pq.pop();
        }
        sum = sum - pq.top().first;
        del[pq.top().second] = true;
        pq.pop();
        C.emplace_back(max(sum, 0 * 1LL));
    }
    
    for (int i = C.size(); i < M; i++) {
        C.emplace_back(0);
    }
    
    return C;
}