提出 #76452134

---

ソースコード 拡げる

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
typedef long long ll;
#define REP(i, n) for (int i = 0, i##_len = (n); i < i##_len; ++i)
#define FOR(i, a, b) for (int i = (a), i##_len = (b); i <= i##_len; ++i)
#define REV(i, a, b) for (int i = (a); i >= (b); --i)
int main() {
  int N,K,M;
  cin >> N >> K >> M; 

  multiset<int> gem;
  map<int,int> mp; // mp[C] := 種類Ciで価値最大のもの

  REP(i,N) {
    int C,V;
    cin >> C >> V;
    if (mp[C] < V) {
      mp[C] = V;
    }
    gem.insert(V);
  }


  vector<int> take_gem;
  for(auto p:mp) {
    take_gem.push_back(p.second);
  }
  sort(take_gem.begin(),take_gem.end());

  ll ans = 0;

  int take_gem_num = take_gem.size();
  for(int i = take_gem_num-1;i >= take_gem_num-M;i--) {
    ans += take_gem[i];
    auto p = gem.find(take_gem[i]);
    gem.erase(p);
  }

  int rem = K - M;

  for(int i = 0;i < rem;i++) {
    auto p = prev(gem.end());
    ans += *p;
    gem.erase(p);
  }

  cout << ans << endl;
}

提出情報

ジャッジ結果