跳跳--openjudge练习题（c++）

描述

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

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

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

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

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

那就请你——会写代码的小跳蛙——写下这个程序.

输入

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

输出

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

样例输入

2
2 1

3
6 3 5

样例输出

5
49

/*跳跳*/
#include<iostream>
#include<algorithm>
#define NUM 300
using namespace std;
static bool cmp(int &m, int &n) {
    return m > n;
}
int main() {
    int n, i,sum=0;
    int a[NUM];
    cin >> n;
    for (i = 0; i < n; i++) {
        cin >> a[i];
    }
    sort(a, a + n, cmp);
    sum =sum+ a[0]*a[0];
    int top = n-1, tail = 0, temp = 0,z=0;
    while (tail<=top){
        if (z == 0) {
            temp = (a[top] - a[tail]) * (a[top] - a[tail]);
            sum =sum+ temp;
            tail++;
            z = 1;
        }
        else {
            temp = (a[top] - a[tail]) * (a[top] - a[tail]);
            sum =sum+ temp;
            top--;
            z = 1;
        }
    }
    cout << sum;
    return 0;
}