Cod sursa(job #585679)

Utilizator Bogdan_tmmTirca Bogdan Bogdan_tmm Data 30 aprilie 2011 10:58:09
Problema Fabrica Scor 0
Compilator cpp Status done
Runda Algoritmiada 2011, Runda Finală, Clasele 10-12 Marime 1.53 kb
#include <algorithm>
#include <queue>

using namespace std;

#define N_MAX 50005
#define INF (1<<30)

int costa[N_MAX],costb[N_MAX];
int finish[N_MAX];
int finish2[N_MAX];

struct cmp
{
	bool operator () (const int a,const int b) const
	{
		return finish[a]>finish[b]||finish[a]==finish[b]&&costa[a]>costa[b];
	}
};

struct cmp2
{
	bool operator () (const int a,const int b) const
	{
		return finish2[a]>finish2[b]||finish2[a]==finish2[b]&&costb[a]>costb[b];
	}
};

priority_queue <int,vector <int>,cmp> PQ1;
priority_queue <int,vector <int>,cmp2 > PQ2;

int n,nra,nrb,i,j;

int dop;
int sol1,sol2;

int main()
{
	freopen("fabrica.in","r",stdin);
	freopen("fabrica.out","w",stdout);
	
	scanf("%d%d%d",&n,&nra,&nrb);
	
	for(i=1;i<=nra;i++)
		scanf("%d",&costa[i]);
	for(i=1;i<=nrb;i++)
		scanf("%d",&costb[i]);
	
	sort(costa+1,costa+nra+1);
	sort(costb+1,costb+nrb+1);
	
	int umplere=0;
	
	for(i=1;i<=nra&&i<=n;i++)
	{
		umplere++;
		PQ1.push(i);
	}
	
	for(i=1;i<=nrb&&i<=n;i++)
		PQ2.push(i);
	
	while(dop<n)
	{
		int nod2=PQ2.top();
		PQ2.pop();
		dop++;
		if(umplere<n||!PQ1.empty())
		{
			int nod=PQ1.top();
			PQ1.pop();
			finish[nod]+=costa[nod];
			
			finish2[nod2]=max(finish[nod],finish2[nod2])+costb[nod2];
			PQ2.push(nod2);
			
			if(umplere<n)
			{
				PQ1.push(nod);
				umplere++;
			}
		}
		
		
	}
	
	for(i=1;i<=nra;i++)
		sol1=max(sol1,finish[i]);
	
	for(i=1;i<=nrb;i++)
		sol2=max(sol2,finish2[i]);
	
	printf("%d %d\n",sol1,sol2);
	
	return 0;
}