跳跳!
题目描述
你是一只小跳蛙,你特别擅长在各种地方跳来跳去。
这一天,你和朋友小 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
样例输入 #1
2
2 1
样例输出 #1
5
样例 #2
样例输入 #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。
解题思路
根据题意不难想到,应该是先排序(例如:从小到大),然后先跳到h[n]h[n]h[n],再从h[n]−>h[1]h[n]->h[1]h[n]−>h[1],接着h[1]−>h[n−1]h[1]->h[n-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;
}