C++贪心——跳跳

题目描述

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

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

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

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

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

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

输入格式

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

输入第二行 n 个正整数，表示第 i块石头的高度 hi​。

输出格式

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

分析：想要消耗的体力最大，就应该选择高度相差大的，将高度按照升序的顺序排序，头尾依次的跳，计算体力

代码：

#include<bits/stdc++.h>
using namespace std;
int a[301];
int main(){
	int n;
	long long t;
	cin >> n;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
	}
	sort(a + 1, a + 1 + n, less<int>());
	int i, j;
	t = a[n] * a[n];
	i = 1;//头
	j = n;//尾
	while (n) {
		t += (a[j] - a[i]) * (a[j] - a[i]);
		j--;
		n--;
		if (i==j||n == 0) {
			cout << t;
			break;
		}
		t += (a[j] - a[i]) * (a[j] - a[i]);
		i++;
		n--;
	}
	return 0;
}