P4995 跳跳！【贪心】

题目描述

你是一只小跳蛙，你特别擅长在各种地方跳来跳去。

这一天，你和朋友小 F 一起出去玩耍的时候，遇到了一堆高矮不同的石头，其中第 iii 块的石头高度为 hih_ihi​，地面的高度是 h0=0h_0 = 0h0​=0。你估计着，从第 iii 块石头跳到第 jjj 块石头上耗费的体力值为 (hi−hj)2(h_i - h_j) ^ 2(hi​−hj​)2，从地面跳到第 iii 块石头耗费的体力值是 (hi)2(h_i) ^ 2(hi​)2。

为了给小 F 展现你超级跳的本领，你决定跳到每个石头上各一次，并最终停在任意一块石头上，并且小跳蛙想耗费尽可能多的体力值。

当然，你只是一只小跳蛙，你只会跳，不知道怎么跳才能让本领更充分地展现。

不过你有救啦！小 F 给你递来了一个写着 AK 的电脑，你可以使用计算机程序帮你解决这个问题，万能的计算机会告诉你怎么跳。

那就请你——会写代码的小跳蛙——写下这个程序，为你 NOIp AK 踏出坚实的一步吧！
输入格式

输入一行一个正整数 nnn，表示石头个数。

输入第二行 nnn 个正整数，表示第 iii 块石头的高度 hih_ihi​。
输出格式

输出一行一个正整数，表示你可以耗费的体力值的最大值。
输入输出样例
输入 #1

2
2 1

输出 #1

5

输入 #2

3
6 3 5

输出 #2

49

说明/提示
样例解释

两个样例按照输入给定的顺序依次跳上去就可以得到最优方案之一。
数据范围

对于 1≤i≤n1 \leq i \leq n1≤i≤n，有 0<hi≤1040 < h_i \leq 10 ^ 40<hi​≤104，且保证 hih_ihi​ 互不相同。

对于 10%10%10% 的数据，n≤3n \leq 3n≤3；

对于 20%20%20% 的数据，n≤10n \leq 10n≤10；

对于 50%50%50% 的数据，n≤20n \leq 20n≤20；

对于 80%80%80% 的数据，n≤50n \leq 50n≤50；

对于 100%100%100% 的数据，n≤300n \leq 300n≤300。

#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstring>
using namespace std;

typedef long long ll;
const int maxn = 2e5 + 10;
int a[maxn];

int main() {
	int n;
	cin >> n;
	for(int i = 1; i <= n; i++) {
		cin >> a[i];
	}
	a[0] = 0;
	ll ans = 0;
	sort(a, a + n + 1);
	int l = 0, r = n;
	while(l != r) {
		ans += pow(a[r] - a[l], 2);
		l++;
		if(l != r) {
			ans += pow(a[r] - a[l], 2);
			r--;
		}
	}
	cout << ans << endl;
	return 0;
}