AGC 018C Coins 贪心(排序)

题意:n个人(n=x+y+z) 每个人有金银铜三种颜色硬币a[i],b[i],c[i]个,问向x个人拿金,y个人拿银,z个人拿铜,只能向每个人要一次,最多能拿到多少硬币? n<=1e5,a[i],b[i],c[i]<=1e9.

先考虑每个人只有两种硬币的情况,按照a[i]-b[i]从小到大排序后,因为j>i时 a[j]+b[i]>a[i]+b[j] 任意一个金都应该选在任意一个银之后才能能最优.

把c累加进ans,(a,b,c)就转换成(a-c,b-c,0) 则问题变为剩下n个选x个金,y个银的最大值.

枚举分界点k(k>=y),从前k个中选最大的y个银,后k+1个中选出最大的x个金.优先队列或者multiset维护.

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+20;
ll f[N],g[N],n,x,y,z;
struct node{
	ll a,b,c;
}a[N];
bool cmp(node a,node b)
{
	return a.a-a.b<b.a-b.b;
}

multiset<ll> s;
int main()
{
	while(cin>>x>>y>>z)
	{
		ll res=0;
		n=x+y+z;
		s.clear();
		for(int i=1;i<=n;i++)
		{
			scanf("%lld%lld%lld",&a[i].a,&a[i].b,&a[i].c);
			res+=a[i].c;
			a[i].a-=a[i].c,a[i].b-=a[i].c,a[i].c=0;
		}
		sort(a+1,a+1+n,cmp);
		ll sum=0;
		for(int i=1;i<=n;i++)
		{
			sum+=a[i].b;
			s.insert(a[i].b);
			if(s.size()>y)
			{
				sum-=*s.begin();
				s.erase(s.begin());
			}
			f[i]=sum;
		}
		s.clear(),sum=0;
		for(int i=n;i>=1;i--)
		{
			sum+=a[i].a;
			s.insert(a[i].a);
			if(s.size()>x)
			{
				sum-=*s.begin();
				s.erase(s.begin());
			}
			g[i]=sum;
		}
		ll ans=0;
		for(int i=y;i<=n-x;i++)
			ans=max(ans,res+f[i]+g[i+1]);
		cout<<ans<<endl;	
	}
	return 0;
}