Submission #1451152

---

Source Code Expand

#include<bits/stdc++.h>

typedef long long LL;  
using namespace std;

#define fillchar(a, x) memset(a, x, sizeof(a))
#define MP make_pair
#define PB push_back
#define endl '\n'

const LL M = (LL)1e9+7;

struct pt{LL a,b,c,id;bool ca,cb,cc;bool pres;};

bool funab(const pt& u,const pt &v)
{
	return ((u.a-u.b) < (v.a-v.b));
}

bool funbc(const pt& u,const pt &v)
{
	return ((u.b-u.c) < (v.b-v.c));
}

bool funac(const pt& u,const pt &v)
{
	return ((u.a-u.c) < (v.a-v.c));
}

int main()
{
//ios_base::sync_with_stdio(0); 
cout.precision(15);
cout.setf(ios::fixed);

	LL x,y,z;
	cin >> x >> y >> z;

	LL n = x+y+z;
	// cout<<n<<endl;
	vector <pt> v(n); 

	for(LL i=0;i<n;i++)
	{
		cin >> v[i].a>> v[i].b >> v[i].c;
		v[i].ca = v[i].cb = v[i].cc = false;
		v[i].pres = true;
	}

	LL cnt = 0;
	LL ans = 0;

	while(x+y+z>0)
	{			
		// cout<<x<<' '<<y<<' '<<z<<endl;

		sort(v.begin(),v.end(),&funab);
		reverse(v.begin(),v.end());
		for(LL i=0,cnt=0;cnt<y;i++)
			if(v[i].pres)
				v[i].cb = true,cnt++;
		for(LL i=n-1,cnt=0;cnt<x;i--)
			if(v[i].pres)
				v[i].ca = true,cnt++;

		sort(v.begin(),v.end(),&funbc);
		reverse(v.begin(),v.end());

		for(LL i=0,cnt=0;cnt<z;i++)
			if(v[i].pres)
				v[i].cc = true,cnt++;
		for(LL i=n-1,cnt=0;cnt<y;i--)
			if(v[i].pres)
				v[i].cb = true,cnt++;
						
		sort(v.begin(),v.end(),&funac);
		reverse(v.begin(),v.end());

		for(LL i=0,cnt=0;cnt<z;i++)
			if(v[i].pres)
				v[i].cc = true,cnt++;
		for(LL i=n-1,cnt=0;cnt<x;i--)
			if(v[i].pres)
				v[i].ca = true,cnt++;




		for(LL i=0;i<n;i++)
		{	
			// cout<<i<<' '<<v[i].pres<<' '<<v[i].ca<<' '<<v[i].cb<<endl;
			if(v[i].pres && (v[i].ca || x==0) && (v[i].cb || y==0) && z>0)
				{ans+=v[i].c,v[i].pres=false,z--;}
			else if(v[i].pres && (v[i].cb || y==0) && (v[i].cc || z==0) && x>0)
				{ans+=v[i].a,v[i].pres=false,x--;}
			else if(v[i].pres && (v[i].ca || x==0) && (v[i].cc || z==0) && y>0)
				{ans+=v[i].b,v[i].pres=false,y--;}
		}	
		// n = x+y+z;	

		// cout << va.size() <<' ' << vb.size() << ' ' << vc.size() << endl;
 	}
	
	cout << ans << endl;	
}

Submission Info

Judge Result