提出 #76464572

---

ソースコード 拡げる

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define LONGMAX 1e18
#define INTMAX 2147483647

int main(void)
{
    ll N, K, M;
    cin >> N >> K >> M;
    vector<ll> C(N, 0), V(N, 0);
    priority_queue<pair<ll, ll>> all;
    priority_queue<ll> remain;
    for (ll i = 0; i < N; i++)
    {
        cin >> C[i] >> V[i];
        all.push({V[i], C[i]});
    }
    set<ll> colors;
    ll ans = 0;
    while (colors.size() < M)
    {
        pair<ll, ll> p = all.top();
        ll value = p.first;
        ll color = p.second;
        all.pop();
        if (colors.count(color) == 0)
        {
            // はじめてみる色なので追加
            colors.insert(color);
            ans += value;
        }
        else
        {
            // 用済みの色ならremainに追加
            remain.push(value);
        }
    }
    // allに残ってる他の要素をremainに移す
    while (all.size() != 0)
    {
        pair<ll, ll> p = all.top();
        ll value = p.first;
        ll color = p.second;
        all.pop();
        remain.push(value);
    }
    // assert(!remain.empty());
    for (ll i = 0, n = K - M; i < n; i++)
    {
        ans += remain.top();
        remain.pop();
    }
    cout << ans << endl;
    return 0;
}

提出情報

コンパイルエラー

./Main.cpp: In function 'int main()':
./Main.cpp:21:26: warning: comparison of integer expressions of different signedness: 'std::set<long long int>::size_type' {aka 'long unsigned int'} and 'long long int' [-Wsign-compare]
   21 |     while (colors.size() < M)
      |            ~~~~~~~~~~~~~~^~~
./Main.cpp:44:12: warning: unused variable 'color' [-Wunused-variable]
   44 |         ll color = p.second;
      |            ^~~~~

ジャッジ結果