Language: C++

#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>

#define ll long long
#define p pair<ll,ll>
#define mp(a,b) make_pair(a,b)
using namespace std;

int main()
{
	int n, k;
	cin >> n >> k;
	vector<p> sushi(n);
	for (int i = 0; i < n; i++)
	{
		ll t, d;
		cin >> t >> d;
		sushi[i] = mp(d,t);
	}
	sort(sushi.begin(), sushi.end());
	reverse(sushi.begin(), sushi.end());
	ll sum = 0;
	set<int> kind;
	priority_queue<ll, vector<ll>, greater<>> doub;
	for (int i = 0; i < k; i++)
	{
		sum += sushi[i].first;
		if (kind.find(sushi[i].second) != kind.end())doub.push(sushi[i].first);
		else kind.insert(sushi[i].second);
	}
	ll ret = sum + kind.size() * kind.size();
	for (int i = k; i < n; i++)
	{
		if (kind.find(sushi[i].second) != kind.end())continue;
		if (doub.empty())break;
		sum += sushi[i].first - doub.top();
		doub.pop();
		kind.insert(sushi[i].second);
		ret = max(ret, sum + (ll)kind.size() * kind.size());
	}
	cout << ret << endl;
	return 0;
}


Compiler output:
a.cc: In function 'int main()':
a.cc:42:26: error: no matching function for call to 'max(long long int&, long long unsigned int)'
   42 |                 ret = max(ret, sum + (ll)kind.size() * kind.size());
      |                       ~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
In file included from /usr/include/c++/14/string:51,
                 from /usr/include/c++/14/bits/locale_classes.h:40,
                 from /usr/include/c++/14/bits/ios_base.h:41,
                 from /usr/include/c++/14/ios:44,
                 from /usr/include/c++/14/ostream:40,
                 from /usr/include/c++/14/iostream:41,
                 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:42:26: note:   deduced conflicting types for parameter 'const _Tp' ('long long int' and 'long long unsigned int')
   42 |                 ret = max(ret, sum + (ll)kind.size() * kind.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,
                 from a.cc:2:
/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:42:26: note:   mismatched types 'std::initializer_list<_Tp>' and 'long long int'
   42 |                 ret = max(ret, sum + (ll)kind.size() * kind.size());
      |                       ~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~