小a与"204"（牛客寒假算法集训营1题目B）

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

时间限制：C/C++ 1秒，
其他语言2秒 空间限制：
C/C++ 262144K，其他语言524288K 64bit IO Format: %lld
题目描述 小a非常喜欢 204 这个数字，因为 ′ a ′ + ′ k ′ = 204 。 现在他有一个长度为 n 的序列，其中只含有 2,0,4这三种数字 设 ai为序列中第 i个数，你需要重新排列这个数列，使得 2 ∑i=1n(ai−ai−1)^2最大
(公式的含义是：每个数与前一个数差的平方的和) 注意：我们默认 a0=0

输入描述: 第一行一个整数 n
接下来一行 n 个整数，第 i 个数表示 ai
输出描述: 输出一个整数，表示 ∑i=1n(ai−ai−1)^2的最大值
示例1
输入2
2 4
输出 20
说明
样例1解释：按 (4,2)排列是最优的，此时 sum=(4−0)^2 + (2−4)^2=20
示例2
输入 3
2 0 4
输出 36
说明 样例2解释：按 (4,0,2)排列是最优的，
此时sum=(4−0)^2 + (0−4)^2 + (2−0)^2=36
示例3
输入
5
2 4 0 2 4
输出
52
备注: 1⩽n⩽105，保证 ai为2/0/4中的数
/*
贪心
排列方式，加一个0，sort一下，先取最小的，再取最大的，之后取第二小的，然后第二大的组成的序列为所求序列，然后直接算OK
*/

#include <stdio.h>
#include <algorithm>
using namespace std;
int a[100005],b[100005];
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        for(int i = 1; i <= n; i++)
        {
            scanf("%d",&a[i]);
        }
        sort(a+1,a+n+1);
        int k = 0;
        for(int i = 0, j = n; i <= n/2; i++,j--)
        {
            b[k++] = a[i];
            if(i != j)
                b[k++] = a[j];
        }
        double sum = 0;
        for(int i = 1; i < k; i++)
        {
            sum += (b[i] - b[i-1])*(b[i] - b[i-1]);
        }
        printf("%.0lf\n",sum);
    }
    return 0;
}