Language: C++

#include <bits/stdc++.h>
using namespace std;
#define rep(i,n) for(ll i=0; i<n; i++)
typedef long long ll;
typedef long long unsigned int llu;
ll mod = 1000000007;

/*
各寿司には、ネタと美味しさのパラメータがある
満足ポイント：美味しさの総和＋ネタの種類の二乗
N個の寿司の中からK個の寿司を選ぶとき、満足ポイントの最大値を求めよ。

k個のお寿司について、p種類のネタを選んだ時の美味しさの総和の最大値をf(p)とする。
求める最大値は、f(p)+p*p
美味しさの高いものからK個お寿司を選ぶとする。(このお寿司の集合をSとする)
この時、このK個のお寿司の種類をxとする。
この場合の美味しさの総和はf(x)となる。

xより小さいiについて、美味しさの総和だけでなく種類も少ないので、i<xなるiについては考えなくていい。
xより大きいiについて、美味しさの総和は確実に減る(または等しいまま)。しかし種類が多くなれば、
満足ポイントは逆転しうる。
x+1,x+2,...,Nまで、以下の手続きで計算できる。
1. Sから、取り除いてもネタの種類が減らないお寿司の中でもっとも美味しさの小さいものを取り出す。
2. Sに存在しないネタを持つ、もっとも美味しさの大きいお寿司を加える。
この操作をNまで繰り返す。

値の挿入および最小値の取り出しは、プライオリティキューを用いる。
プライオリティキューには、取り除いてもネタの種類が減らないお寿司の、美味しさが2番目以降の
お寿司を入れていく。

*/

void solve(){
  int n,k;
  cin >> n >> k;
  vector<pair<int,int>> v;
  rep(i,n){
    int t,d;
    cin >> t >> d;
    v.emplace_back(make_pair(d,t));
  }
  sort(v.rbegin(),v.rend());
  ll sum = 0;
  
  rep(i,k){
    sum += v[i].first;
  }
  
  priority_queue<ll, vector<ll>, greater<ll>> sushi;
  set<ll> sushitabetai;

  rep(i,k){
    if(sushitabetai.find(v[i].second) != sushitabetai.end()){
      sushi.push(v[i].first);
    }
    else{
      sushitabetai.insert(v[i].second);
    }
  }
  ll siz = sushitabetai.size();
  //sum += siz*siz;

  ll ans = sum + siz*siz;
  for(int i = k; i<n; i++){
    cout << "queue top:" << sushi.top() << endl;
    //各お寿司v[k],v[k+1],...,v[n-1]について、
    //ネタが同じならsumには加えない、
    //ネタが同じ出ない時に、プライオリティキューの先頭要素qと比較し、
    //sum-q-siz*siz+(siz+1)*(siz+1)+v[i].first>sumならキューからポップしてsumを更新。
    //ちなみにこの不等式は、2*siz+1+v[i].first>qの時に成立する
    if(sushitabetai.find(v[i].second) == sushitabetai.end() && sushitabetai.size() != k){
      ll q = sushi.top();
      //if(2*siz+1+v[i].first>q){
      //上の評価は実は要らない。なぜなら道中で得られた全てのsumの最大値を計算するため
        sushitabetai.insert(v[i].first);
        sushi.pop();
        sum -= q;
        sum += v[i].first;
        ans = max(ans,sum + sushitabetai.size()*sushitabetai.size());
      //}
    }
  }
  cout << ans << endl;
}

int main(){
  cin.tie(0);
  ios::sync_with_stdio(false);
  solve();
  return 0;
}

Compiler output:
a.cc: In function 'void solve()':
a.cc:79:18: error: no matching function for call to 'max(ll&, long long unsigned int)'
   79 |         ans = max(ans,sum + sushitabetai.size()*sushitabetai.size());
      |               ~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
In file included from /usr/include/c++/14/algorithm:60,
                 from /usr/include/x86_64-linux-gnu/c++/14/bits/stdc++.h:51,
                 from a.cc:1:
/usr/include/c++/14/bits/stl_algobase.h:257:5: note: candidate: 'template<class _Tp> constexpr const _Tp& std::max(const _Tp&, const _Tp&)'
  257 |     max(const _Tp& __a, const _Tp& __b)
      |     ^~~
/usr/include/c++/14/bits/stl_algobase.h:257:5: note:   template argument deduction/substitution failed:
a.cc:79:18: note:   deduced conflicting types for parameter 'const _Tp' ('long long int' and 'long long unsigned int')
   79 |         ans = max(ans,sum + sushitabetai.size()*sushitabetai.size());
      |               ~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
/usr/include/c++/14/bits/stl_algobase.h:303:5: note: candidate: 'template<class _Tp, class _Compare> constexpr const _Tp& std::max(const _Tp&, const _Tp&, _Compare)'
  303 |     max(const _Tp& __a, const _Tp& __b, _Compare __comp)
      |     ^~~
/usr/include/c++/14/bits/stl_algobase.h:303:5: note:   candidate expects 3 arguments, 2 provided
In file included from /usr/include/c++/14/algorithm:61:
/usr/include/c++/14/bits/stl_algo.h:5706:5: note: candidate: 'template<class _Tp> constexpr _Tp std::max(initializer_list<_Tp>)'
 5706 |     max(initializer_list<_Tp> __l)
      |     ^~~
/usr/include/c++/14/bits/stl_algo.h:5706:5: note:   candidate expects 1 argument, 2 provided
/usr/include/c++/14/bits/stl_algo.h:5716:5: note: candidate: 'template<class _Tp, class _Compare> constexpr _Tp std::max(initializer_list<_Tp>, _Compare)'
 5716 |     max(initializer_list<_Tp> __l, _Compare __comp)
      |     ^~~
/usr/include/c++/14/bits/stl_algo.h:5716:5: note:   template argument deduction/substitution failed:
a.cc:79:18: note:   mismatched types 'std::initializer_list<_Tp>' and 'long long int'
   79 |         ans = max(ans,sum + sushitabetai.size()*sushitabetai.size());
      |               ~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~