G - Candy Box (hard version)

思路：用堆贪心，按照总数量为第一关键字，按照f为1的个数为第二关键字，大的在前面，贪心的取，每次更新堆就可以了，push进堆的时候尽量去掉为0的个数

#include<bits/stdc++.h>
using namespace std;
//#define int long long

int mp[2][200010];
int a[200010];
set<int>s;
struct node{
	int xx;
	node(int _X)
	{
		xx=_X;
	}
	bool operator<(const node&w)const
	{
		int a=mp[0][xx]+mp[1][xx];
		int b=mp[0][w.xx]+mp[1][w.xx];
		if(a!=b)
		return a<b;
		return mp[1][xx]<mp[1][w.xx];
	}
};

vector<int>v; 
signed main()
{
	ios::sync_with_stdio(0); 
	int t;
	cin>>t;
	while(t--)
	{
		s.clear();
		int n;
		cin>>n;
		int res=0;
		int res1=0;
		priority_queue<node>q;
		for(int i=1;i<=n;i++)
		{
			int f;
			cin>>a[i]>>f;	
			mp[f][a[i]]++;
			s.insert(a[i]);
		}
		for(auto x:s)
			q.push(node(x));
		int last=0;
		while(q.size())
		{
			auto tx=q.top();
			q.pop();
			int tt=tx.xx;
			int a=mp[0][tt];
			int b=mp[1][tt];
			if(a+b==0)	break;
			if(last!=a+b)
			{
				last=a+b;
				res+=a+b;
				res1+=b;
			}
			else
			{
				if(a==0)
				mp[1][tt]--;
				else
				mp[0][tt]--;
				q.push(node(tt));
			}
		}
		for(int i=1;i<=n;i++)
		mp[0][i]=mp[1][i]=0;	
		cout<<res<<" "<<res1<<endl;
	}
	
	return 0;
}