题目描述
你是一只小跳蛙,你特别擅长在各种地方跳来跳去。
这一天,你和朋友小 F 一起出去玩耍的时候,遇到了一堆高矮不同的石头,其中第 i 块的石头高度为 hi,地面的高度是 ℎ0=0。你估计着,从第 i 块石头跳到第 j 块石头上耗费的体力值为(hi−hj)2,从地面跳到第 i 块石头耗费的体力值是 (hi)^2。
为了给小 F 展现你超级跳的本领,你决定跳到每个石头上各一次,并最终停在任意一块石头上,并且小跳蛙想耗费尽可能多的体力值。
当然,你只是一只小跳蛙,你只会跳,不知道怎么跳才能让本领更充分地展现。
不过你有救啦!小 F 给你递来了一个写着 AK 的电脑,你可以使用计算机程序帮你解决这个问题,万能的计算机会告诉你怎么跳。
那就请你——会写代码的小跳蛙——写下这个程序,为你 NOIp AK 踏出坚实的一步吧!
输入格式
输入一行一个正整数 n,表示石头个数。
输入第二行 n 个正整数,表示第 i 块石头的高度 hi。
输出格式
输出一行一个正整数,表示你可以耗费的体力值的最大值。
输入输出样例
输入
2
2 1
输出
5
输入 #2
3 6 3 5
输出 #2
49
说明/提示
样例解释
两个样例按照输入给定的顺序依次跳上去就可以得到最优方案之一。
数据范围
对于 1≤i≤n,有 0<hi≤104,且保证hi 互不相同。
对于 10%10% 的数据,n≤3;
对于 20%20% 的数据,n≤10;
对于 50%50% 的数据,n≤20;
对于 80%80% 的数据,n≤50;
对于 100%100% 的数据,n≤300。
解:
以下用最简单的方法解释(很大可能不是最优解):
根据题目所说,
(1)两块石头之间的跳消耗的体力值为(hi-hj)^2,要使消耗的体力值尽可能多,则两块石头的落差尽可能大,这样差的平方也最大,即在给石头排完序后从最左边到最右边,再从最右边到最左边,这样“落差”最大。
(2)如果要在一堆数中取出来两个数,使得这两个数差的平方最大,即要取出来这一堆数中最大和最小的两个数。
(3)对于跳过的石头,下次就不再跳到这个石头上来了,所以对于石头的高度数据,每个数据只使用一次
理解到这里,就算差不多了
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int main()
{
long long int n,sum=0;//sum表示消耗的体力值,注意数据范围
int i,j;
cin>>n;
long int a[n+1];//数组存储每个石头的高度,开辟的数组范围要+1
for(i=1;i<=n;i++)
{
cin>>a[i];
}
sort(a+1,a+n+1);//从小到大排序,要寻找最大值和最小值
sum=a[n]*a[n];//sum初始值为青蛙跳到第一块石头上的体力值
i=1;j=n;//i,j相当于指针,指向了这个数组元素的头和尾(当然形象地来说并不是)
while(j>i)//条件,从一边跳到另外一边,不能一块石头跳两次
{//从左边跳到右边,下次从右边跳到左边
sum+=(a[j]-a[i])*(a[j]-a[i]);//消耗的体力值相加
j=j-1;//每次跳过的石头不再跳了
sum+=(a[i]-a[j])*(a[i]-a[j]);
i=i+1;
}
cout<<sum;
return 0;
}