洛谷 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$。

输出格式

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

样例 #1

样例输入 #1

2
2 1

样例输出 #1

5

样例 #2

样例输入 #2

3
6 3 5

样例输出 #2

49

提示

样例解释

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

数据范围

对于 $1\le i\le n$，有 $0<h_i\le 10^4$，且保证 $h_i$ 互不相同。

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

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

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

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

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

解题思路

根据题意不难想到，应该是先排序(例如：从小到大)，然后先跳到$h[n]$，再从$h[n]->h[1]$，接着$h[1]->h[n-1]$，就是这样左右横跳，就可以得到最大值了。

代码

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

int main()
{
    int n;
    cin >> n;
    long long num[n+1], sum = 0;
    for(int i=1;i<=n;i++) cin >> num[i];
    sort(num + 1, num + n + 1);
    sum += num[n]*num[n];
    int cnt = 1, l = 1, r = n;
    while(cnt < n)
    {
        sum += (num[l] - num[r]) * (num[l]-num[r]);
        cnt ++;
        if(cnt>=n) break;
        sum += (num[l] - num[r-1]) * (num[l]-num[r-1]);
        cnt ++;
        l++, r--;
    }
    cout << sum ;
    return 0;
}