P4995 跳跳！

题目描述

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

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

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

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

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

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

输入输出格式

输入格式

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

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

输出格式

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

输入输出样例

输入样例

2
2 1

输出样例

5

输入样例

3
6 3 5

输出样例

说明

样例解释

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

数据范围

对于 $1\le i\le n$，有 KaTeX parse error: Expected 'EOF', got '&' at position 3: 0 &̲lt; h_i \leq 10…，且保证 $h_i$ 互不相同。

对于 $10\%$ 的数据，$n\le 3$；

对于 $20\%$ 的数据，$n\le 10$；

对于 $50\%$ 的数据，$n\le 20$；

对于 $80\%$ 的数据，$n\le 50$；

对于 $100\%$ 的数据，$n\le 300$。

贪心，石头高度排好序，从地面上跳到最高的，然后在跳回最矮的，然后跳回第二高的，然后跳回第二矮的。

#include<iostream>
#include<algorithm>
#include<math.h>
using namespace std;
int main()
{
    int n;
    int num[305];
    long long ans=0;
    
    scanf("%d",&n);
    //从1开始好算结束条件
    for(int i=1;i<=n;i++)
        scanf("%d",&num[i]);
    //排序
    sort(num+1,num+n+1);
    //pre为这一跳的起点，post为终点
    int post=1,pre=n;
    //从地面跳到最高的一块
    ans=pow(num[n],2);
    //接下来一直跳
    while (1)
    {
        //计算这次跳耗费的体力
        ans+=pow(num[pre]-num[post],2);
        //检查有没有跳完所有，终点应该是石头个数的一半
        if(post==int(n/2.0+0.5)) 
            break;
        
        int temp=pre;
        //上一跳的终点，变成起点
        pre=post;
        //当时是从高到低还是从低到高
        if(temp>n/2)
            post=temp-1;
        else
            post=temp+1;
    }
    printf("%lld",ans);
    return 0;
}