Cod sursa(job #649386)

Utilizator indestructiblecont de teste indestructible Data 15 decembrie 2011 22:34:58
Problema Avioane Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.9 kb
#include <stdio.h>
#include <algorithm>
#define NMAX 100005
#define ll long long
using namespace std;
int n,A[NMAX];
ll B[NMAX],act,rez;
inline int better(int dr,int st)
{
	ll act=(ll)A[st]*(dr-st+1);
	if (B[dr]<act)
	{
		B[dr]=act;
		return 1;
	}
	return 0;
}
void rezolva(int dr)
{
	int i,step,st=0;
	for (step=1; step<=dr; step<<=1);
	char ok;
	for (i=0; step; step>>=1)
	{
		ok=1;
		while (ok)
		{
			ok=0;
			if (st+step<=dr && better(dr,st+step))
				st+=step,ok=1;
			if (st-step>0 && better(dr,st-step))
				st-=step,ok=1;
		}
	}
}
int main()
{
	freopen("avioane.in","r",stdin);
	freopen("avioane.out","w",stdout);
	scanf("%d",&n);
	int i;
	for (i=1; i<=n; i++)
		scanf("%d",&A[i]);
	sort(A+1,A+n+1);
	for (i=1; i<=n; i++)
		rezolva(i);
	for (i=1; i<=n; i++)
	{
		act=B[i-1]+(ll)A[i]*(n-i+1);
		if (act>rez)
			rez=act;
	}
	printf("%lld\n",rez);
	return 0;
}