提出 #75278838

---

ソースコード 拡げる

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

/////////////////// メイン ///////////////////

int main () {
  
  //////////////////// 入力 ////////////////////

  int n, k;
  cin >> n >> k;

  vector<int> a(n);
  for (int i=0; i<n; i++) {
    cin >> a.at(i);
  }

  //////////////// 出力変数定義 ////////////////

  long long result = 0;

  //////////////////// 処理 ////////////////////

  // xとして何を選んだら1回の操作で和がいくつ減らせるか、全て求める
  map<long long,int> mp;
  for (int i : a) {
    mp[i]++;
  }

  // 「1回の操作で和がいくつ減らせるか」を全て、大きい順に取り出すpriority_queueに入れる
  priority_queue<long long> pq;
  for (auto [val,num] : mp) {
    pq.emplace(val*num);
  }

  // k回あるいはpqが空になるまで、貪欲に大きい値を捨てる
  for (int i=0; i<k&&!pq.empty(); i++) {
    pq.pop();
  }

  // 残ったものの合計が答え
  while(!pq.empty()) {
    result += pq.top();
    pq.pop();
  }

  //////////////////// 出力 ////////////////////

  cout << result << endl;

  //////////////////// 終了 ////////////////////

  return 0;

}

提出情報

ジャッジ結果