求和（贪心）

链接：https://ac.nowcoder.com/acm/contest/317/B
来源：牛客网
 

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 262144K，其他语言524288K
64bit IO Format: %lld

题目描述

小a非常喜欢204204这个数字，因为′a′+′k′=204′a′+′k′=204。
现在他有一个长度为nn的序列，其中只含有2,0,42,0,4这三种数字
设aiai为序列中第ii个数，你需要重新排列这个数列，使得∑ni=1(ai−ai−1)2∑i=1n(ai−ai−1)2最大(公式的含义是：每个数与前一个数差的平方的和)
注意：我们默认a0=0a0=0

输入描述:

第一行一个整数nn
接下来一行nn个整数，第ii个数表示aiai

输出描述:

输出一个整数，表示∑ni=1(ai−ai−1)2∑i=1n(ai−ai−1)2的最大值

示例1

输入

复制

2
2 4

输出

复制

20

说明

样例1解释：按(4,2)(4,2)排列是最优的，此时sum=(4−0)2+(2−4)2=20sum=(4−0)2+(2−4)2=20

示例2

输入

复制

3
2 0 4

输出

复制

36

说明

样例2解释：按(4,0,2)(4,0,2)排列是最优的，此时sum=(4−0)2+(0−4)2+(2−0)2=36sum=(4−0)2+(0−4)2+(2−0)2=36

示例3

输入

复制

5 
2 4 0 2 4

输出

复制

52

备注:

1⩽n⩽1051⩽n⩽105，保证aiai为2/0/42/0/4中的数

方法一：先排序再求和

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
using namespace std;
typedef long long ll;
const int maxn=1e5+5;
ll a[maxn]; 
ll ans=0;
int main(){
	int n;
	cin>>n;
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
	}
	sort(a+1,a+1+n);
	int l=0,r=n,f=1;
	while(l<=r){
		if(f==0){
			ans+=pow((a[l]-a[r+1]),2);
		}
		ans+=pow((a[l]-a[r]),2);
		l++,r--;
		f=0;
	}
	cout<<ans<<endl;
	return 0;
} 

方法二：分类讨论

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
using namespace std;
typedef long long ll;
ll ans=0;
int main(){
	int n,cnt0=0,cnt2=0,cnt4=0,num;
	cin>>n;
	while(n--){
		scanf("%d",&num);
		if(num==0)  cnt0++;
		if(num==2)  cnt2++;
		if(num==4)  cnt4++;
	}
	while(1)//先排40的情况 
	{
		if(cnt4==0) break;
		cnt4--;
		ans+=16;
		if(cnt0==0) break;
		cnt0--;
		ans+=16;
	}
	if(cnt4==0)//如果0多4少，那么先判断2再判断0 
	{
		while(1){
			if(cnt2==0) break;
			cnt2--;
			ans+=4;
			if(cnt0==0) break;
			cnt0--;
			ans+=4;
		}
	}
	else//如果4多0少先判断2再判断4 
	{
		while(1){
			if(cnt2==0) break;
			cnt2--;
			ans+=4;
			if(cnt4==0) break;
			cnt4--;
			ans+=4; 
		}
	}
	cout<<ans<<endl;
	return 0;
}