小a与"204"

【题目描述】

小a非常喜欢204这个数字，因为′a′+′k′=204。

现在他有一个长度为n的序列，其中只含有2,0,4这三种数字

设ai为序列中第i个数，你需要重新排列这个数列，使得  最大

(公式的含义是：每个数与前一个数差的平方的和)

注意：我们默认a0=0

【输入描述】

第一行一个整数n
接下来一行n个整数，第i个数表示ai

【输出描述】

输出一个整数，表示  的最大值

【样例】

示例1

输入
2
2 4
输出
20
说明
按(4,2)排列是最优的，此时sum=(4−0)^2+(2−4)^2=20

示例2

输入
3
2 0 4
输出
36
说明
按(4,0,2)排列是最优的，此时sum=(4−0)^2+(0−4)^2+(2−0)^2=36

思路：贪心+构造

由于不需要输出方案，因此只需要按照题意进行构造即可

【源代码】

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<utility>
#include<stack>
#include<queue>
#include<deque>
#include<vector>
#include<set>
#include<map>
#define PI acos(-1.0)
#define E 1e-6
#define INF 0x3f3f3f3f
#define N 1000001
#define LL long long
const int MOD=998244353;
const int dx[]={-1,1,0,0};
const int dy[]={0,0,-1,1};
using namespace std;
int a[N];
int main(){
    int n;
    LL sum=0;

    cin>>n;
    for(int i=1;i<=n;i++)
        cin>>a[i];
    
    sort(a,a+n+1);

    for(int i=n;i>n/2;i--)
        sum+=pow(a[i]-a[n-i],2)+pow(a[n-i+1]-a[i],2);

    cout<<sum<<endl;

    return 0;
}